PyTorch使用cpu调用gpu训练的模型

加载模型的时候加上:map_location
model = torch.load(‘xxx.pkl’, map_location=lambda storage, loc: storage)
model = model.module

你可能感兴趣的:(pytorch)