编程 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 22 Python

Python实现Tab自动补全和历史命令管理的方法

Mar 12 Python

Python读写unicode文件的方法

Jul 10 Python

Python构建网页爬虫原理分析

Dec 19 Python

python中多个装饰器的执行顺序详解

Oct 08 Python

Python 实现毫秒级淘宝抢购脚本的示例代码

Sep 16 Python

pandas实现将日期转换成timestamp

Dec 07 Python

解决pytorch-yolov3 train 报错的问题

Feb 18 Python

python如何删除文件、目录

Jun 23 Python

python实现简单的tcp 文件下载

Sep 16 Python

用 Django 开发一个 Python Web API的方法步骤

Dec 03 Python

在Python中实现字典反转案例

Dec 05 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迭代器实现斐波纳契数列的函数

2013/11/12 PHP

PHP实现采集抓取淘宝网单个商品信息

2015/01/08 PHP

PHP的Socket网络编程入门指引

2015/08/11 PHP

php+ajax简单实现全选删除的方法

2016/12/06 PHP

php数组遍历类与用法示例

2019/05/24 PHP

jQuery textarea的长度进行验证

2009/05/06 Javascript

jquery实现div拖拽宽度示例代码

2013/07/31 Javascript

js实现幻灯片效果(基于jquery插件)

2013/11/05 Javascript

用jQuery模拟select下拉框的简单示例代码

2014/01/26 Javascript

node.js中的fs.stat方法使用说明

2014/12/16 Javascript

JavaScript中使用Callback控制流程介绍

2015/03/16 Javascript

JS获取IE版本号与HTML设置IE文档模式的方法

2016/10/09 Javascript

jquery实现tab选项卡切换效果（悬停、下方横线动画位移）

2017/05/05 jQuery

JQuery实现定时刷新功能代码

2017/05/09 jQuery

详解vue-router和vue-cli以及组件之间的传值

2017/07/04 Javascript

AngularJS使用ui-route实现多层嵌套路由的示例

2018/01/10 Javascript

Vue 中axios配置实例详解

2018/07/27 Javascript

vue视频播放插件vue-video-player的具体使用方法

2019/11/08 Javascript

原生JS实现无缝轮播图片

2020/06/24 Javascript

JavaScript this指向相关原理及实例解析

2020/07/10 Javascript

python 读取鼠标点击坐标的实例

2018/12/29 Python

Python实现的栈、队列、文件目录遍历操作示例

2019/05/06 Python

python爬取Ajax动态加载网页过程解析

2019/09/05 Python

python利用opencv实现SIFT特征提取与匹配

2020/03/05 Python

python为Django项目上的每个应用程序创建不同的自定义404页面(最佳答案)

2020/03/09 Python

html5指南-4.使用Geolocation实现定位功能

2013/01/07 HTML / CSS

HTML5 video 事件应用示例

2014/09/11 HTML / CSS

使用phonegap获取位置信息的实现方法

2017/03/31 HTML / CSS

英国最大的割草机购买网站：Just Lawnmowers

2019/11/02 全球购物

得到Class的三个过程是什么

2012/08/10 面试题

高中生家长会演讲稿

2014/01/14 职场文书

工厂仓管员岗位职责范本

2014/07/17 职场文书

毕业生就业推荐表导师评语

2014/12/31 职场文书

四则混合运算教学反思

2016/02/23 职场文书

PHP中->和=>的意思

2021/03/31 PHP

解决python存数据库速度太慢的问题

2021/04/23 Python