编程 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中的输入和输出功能进行读取和写入的教程

Apr 14 Python

在Python程序中操作文件之isatty()方法的使用教程

May 24 Python

python过滤字符串中不属于指定集合中字符的类实例

Jun 30 Python

Python使用matplotlib绘制正弦和余弦曲线的方法示例

Jan 06 Python

python 接口返回的json字符串实例

Mar 27 Python

python筛选出两个文件中重复行的方法

May 31 Python

python Selenium实现付费音乐批量下载的实现方法

Jan 24 Python

深入了解python中元类的相关知识

Aug 29 Python

Python和Anaconda和Pycharm安装教程图文详解

Feb 04 Python

Python实现密钥密码（加解密）实例详解

Apr 26 Python

Pycharm生成可执行文件.exe的实现方法

Jun 02 Python

python collections模块的使用

Oct 16 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中for循环语句的几种变型

2006/11/26 PHP

php 实现进制转换(二进制、八进制、十六进制）互相转换实现代码

2010/10/22 PHP

php表单提交问题的解决方法

2011/04/12 PHP

PHP 的异常处理、错误的抛出及回调函数等面向对象的错误处理方法

2012/12/07 PHP

使用PHP进行微信公众平台开发的示例

2015/08/21 PHP

PHP简单数据库操作类实例【支持增删改查及链式操作】

2016/10/10 PHP

PHP读取文本文件并逐行输出该行使用最多的字符与对应次数的方法

2016/11/25 PHP

laravel框架模型和数据库基础操作实例详解

2020/01/25 PHP

javascript 表单验证常见正则

2009/09/28 Javascript

javascript 获取元素位置的快速方法 getBoundingClientRect()

2009/11/26 Javascript

js获取单选框或复选框值及操作

2012/12/18 Javascript

jQuery中对未来的元素绑定事件用bind、live or on

2014/04/17 Javascript

jQuery实现页面内锚点平滑跳转特效的方法总结

2015/05/11 Javascript

AngularJS动态生成div的ID源码解析

2016/08/29 Javascript

AngularJs基于角色的前端访问控制的实现

2016/11/07 Javascript

脚本div实现拖放功能(两种)

2017/02/13 Javascript

详解JS中的attribute属性

2017/04/25 Javascript

老生常谈Bootstrap媒体对象

2017/07/06 Javascript

layui多iframe页面控制定时器运行的方法

2019/09/05 Javascript

Vue实例的对象参数options的几个常用选项详解

2019/11/08 Javascript

在vue-cli3.0 中使用预处理器 (Sass/Less/Stylus) 配置全局变量操作

2020/08/10 Javascript

[02:36]DOTA2英雄基础教程斯拉克

2013/11/29 DOTA

Scrapy的简单使用教程

2017/10/24 Python

python定时复制远程文件夹中所有文件

2019/04/30 Python

Python实现socket非阻塞通讯功能示例

2019/11/06 Python

jupyter notebook远程访问不了的问题解决方法

2021/01/11 Python

CSS3实现银灰色动画效果的导航菜单代码

2015/09/01 HTML / CSS

html5中svg canvas和图片之间相互转化思路代码

2014/01/24 HTML / CSS

Foot Locker德国官方网站：美国运动服和鞋类零售商

2018/11/01 全球购物

最新的咖啡店创业计划书

2013/12/30 职场文书

档案接收函范文

2014/01/10 职场文书

医学生临床实习自我评价

2014/03/07 职场文书

我的中国梦演讲稿500字

2014/08/19 职场文书

落实八项规定专题民主生活会对照检查材料

2014/09/15 职场文书

浅析JavaScript中的变量提升

2022/06/01 Javascript