Pytorch/Torch：多GPU与单GPU相互修改

在使用Pytorch加载模型时出现这个错误语句。

原因：原本的模型是用两个GPU训练的，而你的电脑只有一个，所以会出错。

解决：

model = torch.load(model_path)

改为：

model = torch.load(model_path, map_location='cuda:0')

如果是4块到两块：就把map_location改为：map_location={'cuda:1': 'cuda:0'}

其余情况以此类推。。。