Use CUDA_VISIBLE_DEVICES
(not “DEVICE”). You have to set it before you launch the program – you can’t do it from within the program.
5 Likes
Use CUDA_VISIBLE_DEVICES
(not “DEVICE”). You have to set it before you launch the program – you can’t do it from within the program.