Hi! I’m adding  os.environ['CUDA_VISIBLE_DEVICES'] = "2"  in my code does not work, the code always select first GPU, however  CUDA_VISIBLE_DEVICES=2 python train.py  works.
I find that  os.environ['CUDA_VISIBLE_DEVICES']  works in this code 65 . I have check my code for multiple times and haven’t got the reason. Who can help me? Thanks a lot.