编程 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抓取手机号归属地信息示例代码

Nov 28 Python

Python实现字符串格式化的方法小结

Feb 20 Python

python win32 简单操作方法

May 25 Python

利用python求解物理学中的双弹簧质能系统详解

Sep 29 Python

Python基于OpenCV实现视频的人脸检测

Jan 23 Python

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

Feb 06 Python

Django中日期处理注意事项与自定义时间格式转换详解

Aug 06 Python

详解python执行shell脚本创建用户及相关操作

Apr 11 Python

PyQt5 对图片进行缩放的实例

Jun 18 Python

pip安装python库的方法总结

Aug 02 Python

tensorflow 实现从checkpoint中获取graph信息

Feb 10 Python

Python中rapidjson参数校验实现

Jul 25 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

使用字符串函数输出整数化的PHP版本号

2006/10/09 PHP

PHP快速按行读取CSV大文件的封装类分享（也适用于其它超大文本文件）

2014/04/10 PHP

php+mysql实现数据库随机重排实例

2014/10/17 PHP

php实现socket推送技术的示例

2017/12/20 PHP

PHP实现基本留言板功能原理与步骤详解

2020/03/26 PHP

JS运行耗时操作的延时显示方法

2010/11/19 Javascript

在IE 浏览器中使用 jquery的fadeIn() 效果英文字符字体加粗

2011/06/02 Javascript

javascript实现时间格式输出FormatDate函数

2015/01/13 Javascript

javascript最基本的函数汇总

2015/06/25 Javascript

每天一篇javascript学习小结（Function对象）

2015/11/16 Javascript

JS清除文本框内容离开在恢复及鼠标离开文本框时触发js的方法

2016/01/12 Javascript

JavaScript jquery及AJAX小结

2016/01/24 Javascript

Bootstrap中glyphicons-halflings-regular.woff字体报404错notfound的解决方法

2017/01/19 Javascript

Bootstrap里的文件分别代表什么意思及其引用方法

2017/05/01 Javascript

AngularJS ng-repeat指令及Ajax的应用实例分析

2017/07/06 Javascript

jQuery滚动条美化插件nicescroll简单用法示例

2018/04/18 jQuery

Node.js进阶之核心模块https入门

2018/05/23 Javascript

vue中v-model的应用及使用详解

2018/06/27 Javascript

Vant的安装和配合引入Vue.js项目里的方法步骤

2018/12/05 Javascript

利用JS响应式修改vue实现页面的input值

2019/09/02 Javascript

Python监控主机是否存活并以邮件报警

2015/09/22 Python

Python多继承顺序实例分析

2018/05/26 Python

对django xadmin自定义菜单的实例详解

2019/01/03 Python

为什么你还不懂得怎么使用Python协程

2019/05/13 Python

关于Numpy数据类型对象(dtype)使用详解

2019/11/27 Python

TensorFlow设置日志级别的几种方式小结

2020/02/04 Python

python + selenium 刷B站播放量的实例代码

2020/06/12 Python

Jmeter调用Python脚本实现参数互相传递的实现

2021/01/22 Python

anaconda安装pytorch1.7.1和torchvision0.8.2的方法(亲测可用)

2021/02/01 Python

独特的礼品和创新的科技产品：The Grommet

2018/02/24 全球购物

英国领先的在线高尔夫设备零售商：Golfgeardirect

2020/12/11 全球购物

制药工程专业应届生求职信

2013/09/24 职场文书

社会实践自我鉴定

2013/11/07 职场文书

物流仓管员岗位职责

2013/12/04 职场文书

会计电算化大学生职业规划书

2014/02/05 职场文书

导游词之重庆钓鱼城

2019/09/19 职场文书