RuntimeError: module must have its parameters and buffers on device cuda:1 (device_ids[0]) but found

解决RuntimeError: module must have its parameters and buffers on device cuda:1 (device_ids[0]) but found one of them on device: cuda:0

在使用多卡训练时，借助nn.DataParallel来指定使用某几张GPU
如使用1，2号

model=nn.DataParallel(model,device_ids=[1,2])
model.to(device)

然后前向传播就会碰到标题的错误
究其原因，问题出现在这一句上：

model.to(device)

这里的device指向的那一块GPU就是模型的参数等要放置的GPU，torch规定：必须把参数放置在nn.DataParallel中参数device_ids[0]上，在这里device_ids=[1,2]，因此我们需要

device=torch.device("cuda:1" )
model.to(device)

然后再运行模型，应该一切okl