RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1.

688 阅读1分钟
出现这样的原因是因为我用了第二块GPU训练,在另一台机器上加载模型的时候,没有了第二块GPU,这时候就要指定用“cuda:0”
torch.load('./modelSubmit/encoder_48.pth.tar',map_location='cuda:0')

然后加载模型就正常了。