-
- Downloads
Fixed accelerator assignement
The CUDA_VISIBLE_DEVICES environement variable is set if a device is specified or we raise an error if it has not been set. Code for accelerator handling has been moved to its own utils script.
The CUDA_VISIBLE_DEVICES environement variable is set if a device is specified or we raise an error if it has not been set. Code for accelerator handling has been moved to its own utils script.