编程 Python

Pytorch GPU显存充足却显示out of memory的解决方式

Posted in Python onJanuary 13, 2020

今天在测试一个pytorch代码的时候显示显存不足，但是这个网络框架明明很简单，用CPU跑起来都没有问题，GPU却一直提示out of memory.

在网上找了很多方法都行不通，最后我想也许是pytorch版本的问题，原来我的pytorch版本是0.4.1，于是我就把这个版本卸载，然后安装了pytorch1.1.0，程序就可以神奇的运行了，不会再有OOM的提示了。虽然具体原因还不知道为何，这里还是先mark一下，具体过程如下：

卸载旧版本pytorch：

conda uninstall pytorch

安装pytorch1.1.0，按照官网上的办法，我的CUDA版本是9.0：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

大功告成!

以上这篇Pytorch GPU显存充足却显示out of memory的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

Pytorch GPU显存充足却显示out of memory的解决方式

- Author -

imaginist233

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐

跟老齐学Python之从格式化表达式到方法

Sep 28 Python

python多线程threading.Lock锁用法实例

Nov 01 Python

python实现给数组按片赋值的方法

Jul 28 Python

Python获取当前页面内所有链接的四种方法对比分析

Aug 19 Python

python实现决策树

Dec 21 Python

详解Python 爬取13个旅游城市，告诉你五一大家最爱去哪玩？

May 07 Python

Django单元测试工具test client使用详解

Aug 02 Python

python多进程重复加载的解决方式

Dec 13 Python

将tensorflow模型打包成PB文件及PB文件读取方式

Jan 23 Python

Python如何安装第三方模块

May 28 Python

Python selenium模拟网页点击爬虫交管12123违章数据

May 26 Python

基于Python和openCV实现图像的全景拼接详细步骤

Oct 05 Python

Python开发之基于模板匹配的信用卡数字识别功能

Jan 13 #Python

python中的itertools的使用详解

Jan 13 #Python

python3读取csv文件任意行列代码实例

Jan 13 #Python

pytorch程序异常后删除占用的显存操作

Jan 13 #Python

Python跑循环时内存泄露的解决方法

Jan 13 #Python

PyTorch使用cpu加载模型运算方式

Jan 13 #Python

Python如何读取文件中图片格式

Jan 13 #Python

You might like

Terran历史背景

2020/03/14 星际争霸

php使用parse_url和parse_str解析URL

2015/02/22 PHP

ThinkPHP5实现作业管理系统中处理学生未交作业与已交作业信息的方法

2016/11/12 PHP

详解PHP中的 input属性（隐藏只读限制）

2017/08/14 PHP

修改Laravel自带的认证系统的User类的命名空间的步骤

2019/10/15 PHP

javascript web对话框与弹出窗口

2009/02/22 Javascript

Extjs4.0设置Ext.data.Store传参的请求方式(默认为GET)

2013/04/02 Javascript

js对象转json数组的简单实现案例

2014/02/28 Javascript

javascript的事件触发器介绍的实现

2014/06/05 Javascript

jQuery控制TR显示隐藏的几种方法

2014/06/18 Javascript

jQuery的animate函数学习记录

2014/08/08 Javascript

Bootstrap编写一个同时适用于PC、平板、手机的登陆页面

2016/06/30 Javascript

ionic2 tabs 图标自定义实例

2017/03/08 Javascript

vue实现类似淘宝商品评价页面星级评价及上传多张图片功能

2018/10/29 Javascript

JS拖拽排序插件Sortable.js用法实例分析

2019/02/20 Javascript

微信小程序如何实现全局重新加载

2019/06/05 Javascript

javascript如何实现create方法

2019/11/04 Javascript

RxJS在TypeScript中的简单使用详解

2020/04/13 Javascript

js实现移动端图片滑块验证功能

2020/09/29 Javascript

[16:01]夜魇凡尔赛茶话会第二期01：你比划我猜

2021/03/11 DOTA

布同统计英文单词的个数的python代码

2011/03/13 Python

Python 常用string函数详解

2016/05/30 Python

Python的Flask框架中使用Flask-Migrate扩展迁移数据库的教程

2016/06/14 Python

Python的网络编程库Gevent的安装及使用技巧

2016/06/24 Python

Python3实现发送QQ邮件功能（文本）

2017/12/15 Python

python模块之subprocess模块级方法的使用

2019/03/26 Python

关于tf.TFRecordReader()函数的用法解析

2020/02/17 Python

测绘工程系学生的自我评价

2013/11/30 职场文书

企业演讲比赛主持词

2014/03/18 职场文书

桥梁工程专业求职信

2014/04/21 职场文书

个人自荐材料

2014/05/23 职场文书

缓刑人员思想汇报500字

2014/09/12 职场文书

高三复习计划

2015/01/19 职场文书

电影地道战观后感

2015/06/04 职场文书

丧事主持词

2015/07/02 职场文书

Mysql数据库表中为什么有索引却没有提高查询速度

2022/02/24 MySQL