服务器多GPU的情况下，指定特定的GPU做训练

问题描述

pytorch调用GPU时，默认调用GPU0，但为了错峰使用，我们往往需要把程序挂载在其他GPU上

解决方案

import os
os.environ['CUDA_VISIBLE_DEVICES']='1' #此处选择你要用的GPU序号 0，1，2，3 

由于这行代码把默认的CUDA调用的device都修改了，因此后续指定torch.device(“cuda”)不要再画蛇添足地写为torch.device(“cuda:1”)