09/20/2023, 11:59 AM
Thanks for the great framework you implemented. Here's a small issue. I want to run the program on one GPU while the program calls all GPUs. I tried to set
but errors were raised. Any solutions except setting "CUDA_VISIBLE_DIVICE" in the command line?
09/21/2023, 4:02 PM
Hi! We also use the CUDA_VISIBLE_DIVICE flag before the python script, is the easiest option