RuntimeError: Attempting to deserialize object on CUDA device 1 but torch.cuda.device_count() is 1. AI浩 2021-10-26 688 阅读1分钟 出现这样的原因是因为我用了第二块GPU训练,在另一台机器上加载模型的时候,没有了第二块GPU,这时候就要指定用“cuda:0” torch.load('./modelSubmit/encoder_48.pth.tar',map_location='cuda:0') 然后加载模型就正常了。