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'] in this code 289 is work. Do you know why?