os.environ[‘CUDA_VISIBLE_DEVICES‘]无法使用指定GPU的原因

要把
import os
os.os.environ["CUDA_VISIBLE_DEVICES"] = '1'
放在程序import的第一行。因为如果没放在第一行，在import其他文件的时候，其他文件中可能会有import torch。这样的话设置GPU的语句就会失效

参考：https://blog.csdn.net/qq_24367797/article/details/118581970