报错信息:
RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cuda.device_count() is 1. Please use torch.load with map_location to map your storages to an existing device.
原因:加载的模型时用三个GPU训练的,但现在的电脑只有一个GPU,所以报错。
解决方法:
model = torch.load(model_path)
改为
model = torch.load(model_path, map_location='cuda:0')
如果本机有多块GPU要选择,则改为
model = torch.load(model_path, map_location= {'cuda:1': 'cuda:0'})