os.environ[“CUDA_VISIBLE_DEVICES“]设置无效

问题：

os.environ["CUDA_VISIBLE_DEVICES"] = '1'
设定代码运行在第1块显卡，但是该设置无效，程序仍运行在第0块显卡。

解决：

将该代码放到import torch之前