GPU顺序设置不生效

在训练任务时&＃xff0c;默认卡目前在占用&＃xff0c;剩余显存不够新的任务使用&＃xff0c;这个时候考虑换下显卡顺序&＃xff0c;让新任务在训练是将GPU 1当做主卡&＃xff0c;设置使用的是os模块&＃xff0c;设置了个环境变量

os.environ[&＃39;CUDA_VISIBLE_DEVICES&＃39;] &＃61; &＃39;1&＃39;

这个时候发现&＃xff0c;在新任务执行的时候默认还是把 GPU 0 当做主卡&＃xff0c;导致程序运行失败&＃xff1a;

第一次修改&＃xff1a;

看到网上很多博客中提到&＃xff0c;CUDA_VISIBLE_DEVICES 环境变量的设置需要放到导入torch模块之前&＃xff0c;按照修改后&＃xff0c;依然出现上述问题。

第二次修改&＃xff1a;

这次修改直接在代码中修改&＃xff1a;

self.device &＃61; torch.device("cuda:1")