粗暴解决CUDA out of memory的问题


Posted in Python onMay 22, 2021

小渣渣复现大佬project发现GPU跑不动,出现如下报错:

RuntimeError: CUDA out of memory.

看下来最简单粗暴方法就是减少batch_size,慢是慢了不止一点点但至少跑得动了!

补充:Pytorch GPU显存充足却显示out of memory解决办法

今天在测试一个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

大功告成!

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Python中使用cookielib和urllib2配合PyQuery抓取网页信息
Apr 25 Python
Python实现按学生年龄排序的实际问题详解
Aug 29 Python
Python用imghdr模块识别图片格式实例解析
Jan 11 Python
Python+tkinter模拟“记住我”自动登录实例代码
Jan 16 Python
Python实现的括号匹配判断功能示例
Aug 25 Python
Python如何发布程序的详细教程
Oct 09 Python
python Django 创建应用过程图示详解
Jul 29 Python
Python 异常处理Ⅳ过程图解
Oct 18 Python
在Python中实现函数重载的示例代码
Dec 12 Python
Python 将 QQ 好友头像生成祝福语的实现代码
May 03 Python
详细分析Python collections工具库
Jul 16 Python
解决Pytorch dataloader时报错每个tensor维度不一样的问题
May 28 Python
pytorch中的model.eval()和BN层的使用
May 22 #Python
解决Pytorch中关于model.eval的问题
Pytorch 中net.train 和 net.eval的使用说明
May 22 #Python
对PyTorch中inplace字段的全面理解
May 22 #Python
pytorch中F.avg_pool1d()和F.avg_pool2d()的使用操作
May 22 #Python
用python实现监控视频人数统计
Python基础之进程详解
You might like
PHP 出现乱码和Sessions验证问题的解决方法!
2008/12/06 PHP
PHP 高级课程笔记 面向对象
2009/06/21 PHP
怎样使用php与jquery设置和读取cookies
2013/08/08 PHP
PHP面向对象程序设计实例分析
2016/01/26 PHP
PHP实现Session入库/存入redis的方法
2017/05/04 PHP
动手学习无线电
2021/03/10 无线电
javascript TextArea动态显示剩余字符
2008/10/22 Javascript
jQuery 处理网页内容的实现代码
2010/02/15 Javascript
javascript椭圆旋转相册实现代码
2012/01/16 Javascript
Javascript 多物体运动的实现
2014/12/24 Javascript
PHP结合jQuery实现的评论顶、踩功能
2015/07/22 Javascript
JavaScript中利用Array和Object实现Map的方法
2015/07/27 Javascript
js实现可控制左右方向的无缝滚动效果
2016/05/29 Javascript
EasyUI布局 高度自适应
2016/06/04 Javascript
微信小程序 网络请求(post请求,get请求)
2017/01/17 Javascript
JavaScript中动态向表格添加数据
2017/01/24 Javascript
js中new一个对象的过程
2017/02/20 Javascript
EasyUI的DataGrid每行数据添加操作按钮的实现代码
2017/08/22 Javascript
JQuery+Bootstrap 自定义全屏Loading插件的示例demo
2019/07/03 jQuery
vue实现数字动态翻牌的效果(开箱即用)
2019/12/08 Javascript
在react中使用vue的状态管理的方法示例
2020/05/02 Javascript
nestjs中异常过滤器Exceptionfilter的具体使用
2021/02/07 Javascript
[01:21]2018DOTA2亚洲邀请赛4.5采访 打DOTA2也能有女朋友?
2018/04/06 DOTA
[02:23]2018DOTA2亚洲邀请赛趣味视频——反应测试
2018/04/04 DOTA
Python中多线程及程序锁浅析
2015/01/21 Python
Python3从零开始搭建一个语音对话机器人的实现
2019/08/23 Python
Flask框架请求钩子与request请求对象用法实例分析
2019/11/07 Python
英国在线自行车商店:Evans Cycles
2016/09/26 全球购物
汽车工程专业应届生求职信
2013/10/19 职场文书
物业电工岗位职责
2013/11/20 职场文书
违纪检讨书2000字
2014/02/08 职场文书
个人委托书怎么写
2014/04/04 职场文书
职员竞岗演讲稿
2014/05/14 职场文书
2015年度女工工作总结
2015/10/22 职场文书
《分数乘法》教学反思
2016/02/24 职场文书
postgreSQL数据库基础知识介绍
2022/04/12 PostgreSQL