解决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