首页
所有分类
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高级应用实例对比：高效计算大文件中的最长行的长度
Jun 08 Python
浅谈Python 对象内存占用
Jul 15 Python
Python Tkinter实现简易计算器功能
Jan 30 Python
python搭建服务器实现两个Android客户端间收发消息
Apr 12 Python
python基于物品协同过滤算法实现代码
May 31 Python
使用python 打开文件并做匹配处理的实例
Jan 02 Python
python 读取dicom文件,生成info.txt和raw文件的方法
Jan 24 Python
网易2016研发工程师编程题 奖学金（python）
Jun 19 Python
python实现翻转棋游戏（othello）
Jul 29 Python
使用python的turtle绘画滑稽脸实例
Nov 21 Python
Python下利用BeautifulSoup解析HTML的实现
Jan 17 Python
python实现人机猜拳小游戏
Feb 03 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
二级域名(1) 443(1) 默认端口(3) reload(1) 跨域(4) SSL(3) shell(3) 压力测试(1) URL重定向(3) 前后端分离(1)
You might like
PHP下对数组进行排序的函数
2010/08/08 PHP
xml和web特殊字符
2009/04/28 Javascript
Jquery网页出现的乱码问题的三种解决方法
2013/06/30 Javascript
js判断游览器类型及版本号的代码
2014/05/11 Javascript
js获取IP地址的方法小结
2014/07/01 Javascript
Jquery全选与反选点击执行一次的解决方案
2015/08/14 Javascript
JavaScript学习笔记之数组去重
2016/03/23 Javascript
jQuery实现table中的tr上下移动并保持序号不变的实例代码
2016/07/11 Javascript
jQuery判断邮箱格式对错实例代码讲解
2017/04/12 jQuery
canvas简单快速的实现知乎登录页背景效果
2017/05/08 Javascript
js使用generator函数同步执行ajax任务
2017/09/05 Javascript
js定时器实现倒计时效果
2017/11/05 Javascript
详解vue-router传参的两种方式
2018/09/10 Javascript
vue实现的下拉框功能示例
2019/01/29 Javascript
Vue 实现前端权限控制的示例代码
2019/07/09 Javascript
微信小程序自定义联系人弹窗
2020/05/26 Javascript
vue如何在用户要关闭当前网页时弹出提示的实现
2020/05/31 Javascript
前端如何实现动画过渡效果
2021/02/05 Javascript
[02:08]DOTA2英雄基础教程 马格纳斯
2014/01/17 DOTA
[03:55]2014DOTA2国际邀请赛 Fnatic经理采访赢DK在情理之中
2014/07/10 DOTA
python实现带验证码网站的自动登陆实现代码
2015/01/12 Python
安装python时MySQLdb报错的问题描述及解决方法
2018/03/20 Python
Python使用re模块实现信息筛选的方法
2018/04/29 Python
python实现逆序输出一个数字的示例讲解
2018/06/25 Python
Python面向对象之类和对象实例详解
2018/12/10 Python
Python变量类型知识点总结
2019/02/18 Python
Django在admin后台集成TinyMCE富文本编辑器的例子
2019/08/09 Python
解决Django删除migrations文件夹中的文件后出现的异常问题
2019/08/31 Python
Python hashlib常见摘要算法详解
2020/01/13 Python
python如何利用paramiko执行服务器命令
2020/11/07 Python
美国高级工作服品牌：Carhartt
2018/01/25 全球购物
adidas澳大利亚官方网站：adidas Australia
2018/04/15 全球购物
Gloeilampgoedkoop荷兰：在线购买灯泡
2019/02/16 全球购物
总经理职责范文
2013/11/08 职场文书
2014年行政人事工作总结
2014/12/09 职场文书
使用jpa之动态插入与修改(重写save)
2021/11/23 Java/Android