怎么在PyTorch中使用cpu加载模型运算
这篇文章给大家介绍怎么在PyTorch中使用cpu加载模型运算，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。
没gpu没cuda支持的时候加载模型到cpu上计算
将
model = torch.load(path, map_location=lambda storage, loc: storage.cuda(device))
改为
model = torch.load(path, map_location='cpu')
然后删掉所有变量后面的.cuda()方法
关于怎么在PyTorch中使用cpu加载模型运算就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。