PyTorch使用cpu加载模型运算方式

yipeiwu_com5年前 (2020-03-06)Python基础

没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加载模型运算方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。

返回列表

本文实例讲述了Python中MYSQLdb出现乱码的解决方法，分享给大家供大家参考。具体方法如下：一般来说，在使用mysql最麻烦的问题在于乱码。查看mysql的编码：命令：&nb...

本文实例讲述了Python2.7读取PDF文件的方法。分享给大家供大家参考，具体如下：这篇文章示例代码采用的Python版本是2.7，需要下载的插件是PDFMiner，下载地址是htt...

近日，2018年最具就业前景的7大编程语言排行榜出炉了。这次的编程语言排行榜是由CodingDojo(编码道场)发布。在此次的最有“钱”途的编程语言榜单上，Java排名第一，网红编程语言...

前言最近因为工作原因，发现一个Python的时间组件，很好用分享出来！（忘记作者名字了，在这里先感谢了），下面话不多说，来一起看看详细的介绍吧。示例代码： # -*- codi...

本文实例讲述了python实现计算资源图标crc值的方法，分享给大家供大家参考。具体方法如下：实现该功能的关键在于解析资源信息，找到icon的数据，然后计算这些数据的crc 具体实现代...

宜配屋