pytorch cpu调用gpu训练模型

参考：https://blog.youkuaiyun.com/dcrmg/article/details/79503978

 

model = torch.load('../model/model-new4.pkl', map_location=lambda storage, loc: storage)