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

你可能感兴趣的:(pytorch,RuntimeError:,module,must,have,its)