编程 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 subprocess模块学习总结

Mar 13 Python

python合并同类型excel表格的方法

Apr 01 Python

在macOS上搭建python环境的实现方法

Aug 13 Python

python 修改本地网络配置的方法

Aug 14 Python

python用requests实现http请求代码实例

Oct 31 Python

tensorboard 可以显示graph,却不能显示scalar的解决方式

Feb 15 Python

python opencv实现图片缺陷检测（讲解直方图以及相关系数对比法）

Apr 07 Python

Python实现删除某列中含有空值的行的示例代码

Jul 20 Python

Python基于字典实现switch case函数调用

Jul 22 Python

Python自动化办公Excel模块openpyxl原理及用法解析

Nov 05 Python

Python使用grequests并发发送请求的示例

Nov 05 Python

python3爬虫中多线程进行解锁操作实例

Nov 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

2020显卡排行榜天梯图显卡天梯图2020年3月最新版

2020/04/02 数码科技

php命名空间学习详解

2014/02/27 PHP

ThinkPHP中使用ajax接收json数据的方法

2014/12/18 PHP

php eval函数一句话木马代码

2015/05/21 PHP

php上传图片生成缩略图（GD库）

2016/01/06 PHP

php实现文章置顶功能的方法

2016/10/20 PHP

PDO::rollBack讲解

2019/01/29 PHP

5款Javascript颜色选择器

2009/10/25 Javascript

jquery 弹出层实现代码

2009/10/30 Javascript

js复制到剪切板的实例方法

2013/06/28 Javascript

移动手机APP手指滑动切换图片特效附源码下载

2015/11/30 Javascript

易被忽视的js事件问题总结

2016/05/14 Javascript

bootstrap配合Masonry插件实现瀑布式布局

2017/01/18 Javascript

AngularJs 常用的过滤器

2017/05/15 Javascript

jquery鼠标悬停导航下划线滑出效果

2017/09/29 jQuery

element ui里dialog关闭后清除验证条件方法

2018/02/26 Javascript

JS实现生成由字母与数字组合的随机字符串功能详解

2018/05/25 Javascript

详解Ubuntu安装angular-cli遇到的坑

2018/09/08 Javascript

微信小程序实现同一页面取值的方法分析

2019/04/30 Javascript

vue如何自动化打包测试环境和正式环境的dist/test文件

2019/06/06 Javascript

JS函数基本定义与用法示例

2020/01/15 Javascript

微信小程序实现组件顶端固定或底端固定效果(不随滚动而滚动)

2020/04/09 Javascript

在JavaScript中查找字符串中最长单词的三种方法(推荐)

2021/01/18 Javascript

[05:36]DOTA2 2015国际邀请赛中国区预选赛第四日TOP10

2015/05/29 DOTA

python使用新浪微博api上传图片到微博示例

2014/01/10 Python

python 处理微信对账单数据的实例代码

2019/07/19 Python

python TCP包注入方式

2020/05/05 Python

keras多显卡训练方式

2020/06/10 Python

利用CSS3实现炫酷的飞机起飞动画

2016/09/17 HTML / CSS

在HTML5 Canvas中放入图片和保存为图片的方法

2014/05/03 HTML / CSS

StubHub中国：购买和出售全球活动门票

2020/01/01 全球购物

.net面试题

2015/12/22 面试题

给老婆大人的检讨书

2014/02/24 职场文书

酒店端午节活动方案

2014/08/26 职场文书

反对形式主义、官僚主义、享乐主义和奢靡之风整改措施

2014/09/17 职场文书

公司人事管理制度

2015/08/05 职场文书