怎么在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加载模型运算就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。