Kai Mao comments

Repositories
Issues
Comments

Results 4 comments of


                                            Kai Mao

CUDA_VISIBLE_DEVICES not working?

Same question~

CUDA_VISIBLE_DEVICES not working?

> I write os.environ["CUDA_VISIBLE_DEVICES"]="1" before line 45,it works,I don't know it is right or wrong Hi, I wonder which .py file did you change ？

CUDA_VISIBLE_DEVICES not working?

I modified 'os.environ["CUDA_VISIBLE_DEVICES"] = f"{MPI.COMM_WORLD.Get_rank() % GPUS_PER_NODE}"' in 'dist_util.py' at line 27 to 'os.environ["CUDA_VISIBLE_DEVICES"]="1"' in order to specify the GPU for training.

Where are test images(c256/inet256/p256) ?

Same question~