编程 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中文件I/O高效操作处理的技巧分享

Feb 04 Python

django使用xlwt导出excel文件实例代码

Feb 06 Python

ubuntu安装mysql pycharm sublime

Feb 20 Python

在pandas中一次性删除dataframe的多个列方法

Apr 10 Python

Pandas分组与排序的实现

Jul 23 Python

Django 静态文件配置过程详解

Jul 23 Python

Python代理IP爬虫的新手使用教程

Sep 05 Python

带你彻底搞懂python操作mysql数据库(cursor游标讲解)

Jan 06 Python

pyftplib中文乱码问题解决方案

Jan 11 Python

python 实现线程之间的通信示例

Feb 14 Python

python torch.utils.data.DataLoader使用方法

Apr 02 Python

python使用列表的最佳方案

Aug 12 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

在PHP3中实现SESSION的功能（三）

2006/10/09 PHP

用php实现让页面只能被百度gogole蜘蛛访问的方法

2009/12/29 PHP

PHP投票系统防刷票判断流程分析

2012/02/04 PHP

解析php中用PHPMailer来发送邮件的示例(126.com的例子)

2013/06/24 PHP

php给每个段落添加空格的方法

2015/03/20 PHP

Linux系统下PHP-FPM的安装和配置教程

2015/08/17 PHP

jquery实现鼠标滑过后动态图片提示效果实例

2015/08/10 Javascript

比较常见的javascript中定义函数的区别

2015/11/09 Javascript

javascript实现dom元素可拖动

2016/03/21 Javascript

AngularJS中$apply方法和$watch方法用法总结

2016/12/13 Javascript

Ionic3实现图片瀑布流布局

2017/08/09 Javascript

理解nodejs的stream和pipe机制的原理和实现

2017/08/12 NodeJs

最全正则表达式总结：验证QQ号、手机号、Email、中文、邮编、身份证、IP地址等

2017/08/16 Javascript

vue 点击按钮实现动态挂载子组件的方法

2018/09/07 Javascript

Vux+Axios拦截器增加loading的问题及实现方法

2018/11/08 Javascript

JS实现电商商品展示放大镜特效

2020/01/07 Javascript

vue项目在webpack2实现移动端字体自适配功能

2020/06/02 Javascript

基于Vant UI框架实现时间段选择器

2020/12/24 Javascript

JavaScript Dom实现轮播图原理和实例

2021/02/19 Javascript

Python中使用HTMLParser解析html实例

2015/02/08 Python

Python的randrange()方法使用教程

2015/05/15 Python

讲解Python中fileno()方法的使用

2015/05/24 Python

Python单元测试简单示例

2018/07/03 Python

对python中GUI,Label和Button的实例详解

2019/06/27 Python

深入了解Django View（视图系统）

2019/07/23 Python

Python tkinter之ComboBox(下拉框)的使用简介

2021/02/05 Python

GafasWorld哥伦比亚：网上购买眼镜

2017/11/28 全球购物

土木工程专业大学毕业生求职信

2013/10/13 职场文书

医学专业五年以上个人求职信

2013/12/03 职场文书

2014年最新学习全国两会精神心得

2014/03/17 职场文书

市政管理求职信范文

2014/05/07 职场文书

信用卡工资证明格式

2014/09/13 职场文书

单位租房协议范本

2014/12/03 职场文书

SpringBoot使用ip2region获取地理位置信息的方法

2022/06/21 Java/Android

windows11选中自动复制怎么开启? Win11自动复制所选内容的方法

2022/07/23 数码科技

源码安装apache脚本部署过程详解

2022/09/23 Servers