编程 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和shell实现的校验IP地址合法性脚本分享

Oct 23 Python

使用Python压缩和解压缩zip文件的教程

May 06 Python

Python出现segfault错误解决方法

Apr 16 Python

Python反射用法实例简析

Dec 22 Python

在CentOS6上安装Python2.7的解决方法

Jan 09 Python

Django页面数据的缓存与使用的具体方法

Apr 23 Python

Python 20行简单实现有道在线翻译的详解

May 15 Python

树莓派使用python-librtmp实现rtmp推流h264的方法

Jul 22 Python

将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例

Jan 04 Python

详解PyQt5信号与槽的几种高级玩法

Mar 24 Python

Pycharm 使用 Pipenv 新建的虚拟环境(图文详解)

Apr 16 Python

Python绘制动态水球图过程详解

Jun 03 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

解决file_get_contents无法请求https连接的方法

2013/12/17 PHP

浅谈php fopen下载远程文件的函数

2016/11/18 PHP

有效的捕获JavaScript焦点的方法小结

2009/10/08 Javascript

把JavaScript代码改成ES6语法不完全指南(分享)

2017/09/10 Javascript

vue中v-model动态生成的实例详解

2017/10/27 Javascript

form表单数据封装成json格式并提交给服务器的实现方法

2017/12/14 Javascript

vue踩坑记-在项目中安装依赖模块npm install报错

2019/04/02 Javascript

微信小程序获取用户绑定手机号方法示例

2019/07/21 Javascript

nodejs和react实现即时通讯简易聊天室功能

2019/08/21 NodeJs

JavaScript 事件代理需要注意的地方

2020/09/08 Javascript

浅谈js数组splice删除某个元素爬坑

2020/10/14 Javascript

微信小程序中target和currentTarget的区别小结

2020/11/06 Javascript

详解Vue3 Teleport 的实践及原理

2020/12/02 Vue.js

[50:58]2018DOTA2亚洲邀请赛 4.1 小组赛 B组 Mineski vs EG

2018/04/03 DOTA

python之模拟鼠标键盘动作具体实现

2013/12/30 Python

Python使用py2exe打包程序介绍

2014/11/20 Python

python安装Scrapy图文教程

2017/08/14 Python

对Python中DataFrame按照行遍历的方法

2018/04/08 Python

解决tensorflow1.x版本加载saver.restore目录报错的问题

2018/07/26 Python

Django REST framework 单元测试实例解析

2019/11/07 Python

Python对称的二叉树多种思路实现方法

2020/02/28 Python

Python如何爬取51cto数据并存入MySQL

2020/08/25 Python

Python使用Selenium模拟浏览器自动操作功能

2020/09/08 Python

CSS3动画之利用requestAnimationFrame触发重新播放功能

2019/09/11 HTML / CSS

党员的自我评价范文

2014/01/02 职场文书

高校教师自荐信范文

2014/03/13 职场文书

班级旅游计划书

2014/05/03 职场文书

体育系毕业生自荐信

2014/06/28 职场文书

2014年化验员工作总结

2014/11/18 职场文书

胡雪岩故居导游词

2015/02/06 职场文书

财务工作个人总结

2015/02/27 职场文书

材料采购员岗位职责

2015/04/03 职场文书

劳动仲裁撤诉申请书

2015/05/18 职场文书

学术会议领导致辞

2015/07/29 职场文书

导游词之南京汤山温泉

2019/11/26 职场文书

MySQL学习必备条件查询数据

2022/03/25 MySQL