编程 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的Django框架测试驱动开发的教程

Apr 22 Python

python实现百度语音识别api

Apr 10 Python

python+openCV利用摄像头实现人员活动检测

Jun 22 Python

Python调用C语言的实现

Jul 26 Python

使用python实现画AR模型时序图

Nov 20 Python

Python创建空列表的字典2种方法详解

Feb 13 Python

浅谈Python的方法解析顺序(MRO)

Mar 05 Python

python实现超级马里奥

Mar 18 Python

python根据完整路径获得盘名/路径名/文件名/文件扩展名的方法

Apr 22 Python

opencv-python的RGB与BGR互转方式

Jun 02 Python

Python实现计算图像RGB均值方式

Jun 04 Python

python中函数返回多个结果的实例方法

Dec 16 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

You might like

PHP通过插入mysql数据来实现多机互锁实例

2014/11/05 PHP

php创建、获取cookie及基础要点分析

2015/01/26 PHP

简单谈谈PHP中的trait

2017/02/25 PHP

根据判断浏览器类型屏幕分辨率自动调用不同CSS的代码

2007/02/22 Javascript

jquery ajax的success回调函数中实现按钮置灰倒计时

2013/11/19 Javascript

修改或扩展jQuery原生方法的代码实例

2015/01/13 Javascript

JavaScript日期时间与时间戳的转换函数分享

2015/01/31 Javascript

javascript中alert()与console.log()的区别

2015/08/26 Javascript

关于JS 预解释的相关理解

2016/06/28 Javascript

js 博客内容进度插件详解

2017/02/19 Javascript

vue深入解析之render function code详解

2017/07/18 Javascript

浅谈Angular2 模块懒加载的方法

2017/10/04 Javascript

javascript实现获取一个日期段内每天不同的价格(计算入住总价格)

2018/02/05 Javascript

NodeJS父进程与子进程资源共享原理与实现方法

2018/03/16 NodeJs

vue click.stop阻止点击事件继续传播的方法

2018/09/04 Javascript

vue init webpack 建vue项目报错的解决方法

2018/09/29 Javascript

JavaScript函数Call、Apply原理实例解析

2020/02/17 Javascript

webpack中的模式(mode)使用详解

2020/02/20 Javascript

python函数中return后的语句一定不会执行吗？

2017/07/06 Python

简单实现python收发邮件功能

2018/01/05 Python

pyqt5自定义信号实例解析

2018/01/31 Python

Python构建图像分类识别器的方法

2019/01/12 Python

python调用Matplotlib绘制分布点图

2019/10/18 Python

重构Python代码的六个实例

2020/11/25 Python

详解H5本地储存Web Storage

2017/07/03 HTML / CSS

中国酒类在线零售网站：酒仙网

2016/08/20 全球购物

京东港澳售：京东直邮港澳台

2018/01/31 全球购物

任意存：BOXFUL

2018/05/21 全球购物

贝斯特韦斯特酒店集团官网：Best Western

2019/01/03 全球购物

2014入党积极分子破除“四风”思想汇报

2014/09/14 职场文书

幼儿园2014年度工作总结

2014/11/10 职场文书

2015年医务科工作总结范文

2015/05/26 职场文书

个人合作协议范本

2015/08/06 职场文书

户外拓展训练感想

2015/08/07 职场文书

python绘图subplots函数使用模板的示例代码

2021/04/30 Python

整理Python中常用的conda命令操作

2021/06/15 Python