Skip to content
Snippets Groups Projects
Commit d711a4fa authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

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.
parent 2a7513e8
No related branches found
No related tags found
1 merge request!4Moved code to lightning
Pipeline #72545 canceled
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment