指定一张显卡运行出现RuntimeError: binary_op(): expected both inputs to be on same device问题

问题描述：
在命令行指定第6张显卡运行程序，代码运行时出现如下错误

RuntimeError: binary_op(): expected both inputs to be on same device, but input a is on cuda:6 and input b is on cuda:0

经过分析发现：
问题出在类似这样一行代码c= a * torch.Tensor(b).cuda()，最开始使用device = torch.device('cuda:6'),model.to(device)来指定显卡运行，而在将a和b使用的cuda号输出观察发现，其中a是因为直接通过函数传过来的，所以它是在cuda:6上，但是b因为一开始是数组形式，要想跟a进行乘积，需要转换成跟a一样的形式即torch.cuda.TensorFloat，而我使用torch.Tensor(b).cuda()将其直接转换出现一个问题：只要直接在Tensor后用.cuda()的，默认都运行在第0块显卡，即cuda:0

解决办法，亲测有效：

# 训练开头添加
np.random.seed(seed) 
torch.manual_seed(seed)
cudnn.benchmark = True

# 注释掉
'''device = torch.device('cuda:6')
model.to(device)
images, labels = images.to(device), labels.to(device)
'''

# 多GPU运行添加
model = nn.DataParallel(model).cuda()
# 单卡运行添加
model =model.cuda()

# 输入使用Variable代码添加
images = Variable(images)
labels = Variable(labels.cuda())

你的赞就是我的动力，记得点个赞哟(^o)/~