在训练任务时，默认卡目前在占用，剩余显存不够新的任务使用，这个时候考虑换下显卡顺序，让新任务在训练是将GPU 1当做主卡，设置使用的是os模块，设置了个环境变量
os.environ['CUDA_VISIBLE_DEVICES'] = '1' 这个时候发现，在新任务执行的时候默认还是把 GPU 0 当做主卡，导致程序运行失败：
第一次修改：
看到网上很多博客中提到，CUDA_VISIBLE_DEVICES 环境变量的设置需要放到导入torch模块之前，按照修改后，依然出现上述问题。
第二次修改：
这次修改直接在代码中修改：
self.device = torch.device("cuda:1")