编程 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和shell实现的校验IP地址合法性脚本分享

Oct 23 Python

python脚本监控Tomcat服务器的方法

Jul 06 Python

python删除字符串中指定字符的方法

Aug 13 Python

在PyCharm下打包*.py程序成.exe的方法

Nov 29 Python

Python英文文本分词(无空格)模块wordninja的使用实例

Feb 20 Python

python3 小数位的四舍五入(用两种方法解决round 遇5不进)

Apr 11 Python

OpenCV-Python 摄像头实时检测人脸代码实例

Apr 30 Python

Python实现爬取亚马逊数据并打印出Excel文件操作示例

May 16 Python

Django实现发送邮件功能

Jul 18 Python

PyQt5 closeEvent关闭事件退出提示框原理解析

Jan 08 Python

自定义Django默认的sitemap站点地图样式

Mar 04 Python

Python 中Operator模块的使用

Jan 30 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

咖啡冲泡指南咖啡有哪些制作方式单品咖啡意式咖啡

2021/03/06 冲泡冲煮

php中调用其他系统http接口的方法说明

2014/02/28 PHP

IIS下PHP的三种配置方式对比

2014/11/20 PHP

PHP记录和读取JSON格式日志文件

2016/07/07 PHP

PHP 7.0新增加的特性介绍

2017/06/08 PHP

php微信扫码支付 php公众号支付

2019/03/24 PHP

jquery uaMatch源代码

2011/02/14 Javascript

js实现的类似于asp数据字典的数据类型代码实例

2014/09/03 Javascript

javascript实现类似java中getClass()得到对象类名的方法

2015/07/27 Javascript

实例详解jQuery结合GridView控件的使用方法

2016/01/04 Javascript

全面解析标签页的切换方式

2016/08/21 Javascript

JavaScript组合模式学习要点

2016/08/26 Javascript

jQuery根据ID、CLASS、等获取对象的实例

2016/12/04 Javascript

Javascript中的async awai的用法

2017/05/17 Javascript

浅谈Vue.js 组件中的v-on绑定自定义事件理解

2017/11/17 Javascript

VUE异步更新DOM - 用$nextTick解决DOM视图的问题

2020/11/06 Javascript

使用python分析git log日志示例

2014/02/27 Python

python读取浮点数和读取文本文件示例

2014/05/06 Python

python通过pip更新所有已安装的包实现方法

2017/05/19 Python

python生成随机图形验证码详解

2017/11/08 Python

Python中传递值和传递引用的区别解析

2018/02/22 Python

Python SQL查询并生成json文件操作示例

2018/08/17 Python

python实现汉诺塔算法

2021/03/01 Python

Python里字典的基本用法(包括嵌套字典)

2019/02/27 Python

Django在pycharm下修改默认启动端口的方法

2019/07/26 Python

在Python中使用MongoEngine操作数据库教程实例

2019/12/03 Python

pytorch Dataset,DataLoader产生自定义的训练数据案例

2021/03/03 Python

HTML5学习笔记之html5与传统html区别

2016/01/06 HTML / CSS

HTML5获取当前地理位置并在百度地图上展示的实例

2020/07/10 HTML / CSS

苏宁红孩子母婴商城：redbaby

2017/02/12 全球购物

美国正宗奢华复古手袋、珠宝及配饰网站：What Goes Around Comes Around

2018/07/21 全球购物

在校生汽车维修实习自我鉴定

2013/09/19 职场文书

毕业生简单求职信

2013/11/19 职场文书

团队精神口号

2014/06/06 职场文书

投资入股合作协议书

2014/10/28 职场文书

新员工入职感言范文！

2019/07/04 职场文书