pytorch cpu调用gpu训练模型

参考：https://blog.csdn.net/dcrmg/article/details/79503978

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