RuntimeError: Attempting to deserialize object on CUDA device 2 解决办法

该问题通常由于原实验为分布式环境，因此在操作时产生了错误的值传递，要解决该问题只需要定位到对应的device变量或model变量处，然后做出以下更改：

#映射到cpu

model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location='cpu').to('cpu')

#映射到指定GPU

model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location=lambda storage, loc: storage.cuda()).to('cuda')

#简化版GPU映射

model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location='cuda:0').to('cuda:0')


#动态映射

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = torch.load('AoM-ckpt/Twitter2015/AoM.pt', map_location=device).to(device)