Mentions légales du service

Skip to content

#86 specify gpu devices

Julien Rabault requested to merge #86-specify-GPU-devices into master

#86 (closed)

For GPU use I've made some modifications:

  • I removed use_cuda
  • Now can have several variables cuda_device like '0' (use on gpu 0) 'cpu' (use cpu) or '0,2,3' (use gpu 0, 2 and 3) by simply specifying the gpu(s) you want to use.

Merge request reports