首页
所有分类
TAGS
文字工具 EMOJI BIBLE
编程 Python

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

Posted in Python onJanuary 13, 2020

没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加载模型运算方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

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

- Author -

ngy321

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐
用Python的线程来解决生产者消费问题的示例
Apr 02 Python
python通过自定义isnumber函数判断字符串是否为数字的方法
Apr 23 Python
浅谈python numpy中nonzero()的用法
Apr 02 Python
Jupyter notebook远程访问服务器的方法
May 24 Python
提升Python程序性能的7个习惯
Apr 14 Python
python协程gevent案例 爬取斗鱼图片过程解析
Aug 27 Python
python实现银行管理系统
Oct 25 Python
TensorFlow实现自定义Op方式
Feb 04 Python
python退出循环的方法
Jun 18 Python
python 元组和列表的区别
Dec 30 Python
python+selenium+chrome实现淘宝购物车秒杀自动结算
Jan 07 Python
Python中tkinter的用户登录管理的实现
Apr 22 Python
Python如何读取文件中图片格式
Jan 13 #Python
详解python破解zip文件密码的方法
Jan 13 #Python
PyTorch 随机数生成占用 CPU 过高的解决方法
Jan 13 #Python
python批量处理txt文件的实例代码
Jan 13 #Python
Python hashlib常见摘要算法详解
Jan 13 #Python
Pytorch释放显存占用方式
Jan 13 #Python
pytorch 实现删除tensor中的指定行列
Jan 13 #Python
vue-router-link(1) TimePickerView(1) 监听(2) Yolov5(1) Jmerte(1) 项目打包(2) watch(1) vue-cli3.x(1) KVM(1) scss(1)
You might like
php 静态页面中显示动态内容
2009/08/14 PHP
PHPWind 发帖回帖Api PHP版打包下载
2010/02/08 PHP
PHP实现QQ、微信和支付宝三合一收款码实例代码
2018/02/19 PHP
基于PHP的登录和注册的功能的实现
2020/08/06 PHP
cnblogs TagCloud基于jquery的实现代码
2010/06/11 Javascript
js与jQuery实现checkbox复选框全选/全不选的方法
2016/01/05 Javascript
AngularJS中的指令全面解析(必看)
2016/05/20 Javascript
关于在Servelet中如何获取当前时间的操作方法
2016/06/28 Javascript
jQueryUI 拖放排序遇到滚动条时有可能无法执行排序的小bug及解决方案
2016/12/19 Javascript
angularJS模态框$modal实例代码
2017/05/27 Javascript
js调用设备摄像头的方法
2018/07/19 Javascript
vue-cli中安装方法(图文详细步骤)
2018/12/12 Javascript
JS中注入eval, Function等系统函数截获动态代码
2019/04/03 Javascript
Ant Design moment对象和字符串之间的相互转化教程
2020/10/27 Javascript
[10:34]DOTA2上海特级锦标赛全纪录
2016/03/25 DOTA
Python操作sqlite3快速、安全插入数据（防注入）的实例
2014/04/26 Python
用python删除java文件头上版权信息的方法
2014/07/31 Python
python Django批量导入不重复数据
2016/03/25 Python
Python中 map()函数的用法详解
2018/07/10 Python
python利用小波分析进行特征提取的实例
2019/01/09 Python
Flask框架单例模式实现方法详解
2019/07/31 Python
Window10下python3.7 安装与卸载教程图解
2019/09/30 Python
基于Python检测动态物体颜色过程解析
2019/12/04 Python
Python urlopen()和urlretrieve()用法解析
2020/01/07 Python
利用pyecharts读取csv并进行数据统计可视化的实现
2020/04/17 Python
python 通过exifread读取照片信息
2020/12/24 Python
全球知名鞋履品牌授权零售商：Journeys
2016/09/17 全球购物
检察官就职演讲稿
2014/01/13 职场文书
总经理文秘岗位职责
2014/02/03 职场文书
信用卡结清证明怎么写
2014/09/13 职场文书
大学生学习计划书
2014/09/15 职场文书
员工升职自荐信
2015/03/27 职场文书
2015年推普周活动方案
2015/05/06 职场文书
2015小学五年级班主任工作总结
2015/05/21 职场文书
如何做好工作总结！
2019/04/10 职场文书
Vue中foreach数组与js中遍历数组的写法说明
2021/06/05 Vue.js