编程 Python

粗暴解决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

大功告成！

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

粗暴解决CUDA out of memory的问题

- Author -

李橡树

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

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