首页
所有分类
TAGS
文字工具 EMOJI BIBLE
编程 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 调用VC++的动态链接库(DLL)
Sep 06 Python
django框架如何集成celery进行开发
May 24 Python
python正则表达式及使用正则表达式的例子
Jan 22 Python
PyQt5每天必学之关闭窗口
Apr 19 Python
Win10下python3.5和python2.7环境变量配置教程
Sep 18 Python
Python网页正文转换语音文件的操作方法
Dec 09 Python
弄懂这56个Python使用技巧(轻松掌握Python高效开发)
Sep 18 Python
关于keras.layers.Conv1D的kernel_size参数使用介绍
May 22 Python
python实现经纬度采样的示例代码
Dec 10 Python
Python实现Appium端口检测与释放的实现
Dec 31 Python
python爬虫实现爬取同一个网站的多页数据的实例讲解
Jan 18 Python
django学习之ajax post传参的2种格式实例
May 14 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基础之进程详解
capture(1) Flutter(2) self(1) SpringBoot(22) 数据重复(1) Apache(17) uniapp(4) Navigator(1) element-china-area-data(1) tabBar(1)
You might like
无法在发生错误时创建会话，请检查 PHP 或网站服务器日志，并正确配置 PHP 安装最快的解决办法
2010/08/01 PHP
使用PHP计算两个路径的相对路径
2013/06/14 PHP
PHP中Session和Cookie是如何操作的
2015/10/10 PHP
关于扩展 Laravel 默认 Session 中间件导致的 Session 写入失效问题分析
2016/01/08 PHP
Zend Framework教程之资源(Resources)用法实例详解
2016/03/14 PHP
PHP中TP5 上传文件的实例详解
2017/07/31 PHP
php+redis消息队列实现抢购功能
2018/02/08 PHP
PHP fopen函数用法实例讲解
2019/02/15 PHP
最流行的Node.js精简型和全栈型开发框架介绍
2015/02/26 Javascript
谈谈JavaScript类型系统之Math
2016/01/06 Javascript
JS中使用变量保存arguments对象的方法
2016/06/03 Javascript
JS实现数字格式千分位相互转换方法
2016/08/01 Javascript
jQuery实现智能判断固定导航条或侧边栏的方法
2016/09/04 Javascript
js调用屏幕宽度的简单方法
2016/11/14 Javascript
jquery封装插件时匿名函数形参和实参的写法解释
2017/02/14 Javascript
vue项目中导入swiper插件的方法
2018/01/30 Javascript
使用vue-infinite-scroll实现无限滚动效果
2018/06/22 Javascript
ES6知识点整理之Proxy的应用实例详解
2019/04/16 Javascript
深入了解JavaScript 私有化
2019/05/30 Javascript
对于Python装饰器使用的一些建议
2015/06/03 Python
python中引用与复制用法实例分析
2015/06/04 Python
Python计算斗牛游戏概率算法实例分析
2017/09/26 Python
Python学习笔记之if语句的使用示例
2017/10/23 Python
django表单实现下拉框的示例讲解
2018/05/29 Python
python实现将读入的多维list转为一维list的方法
2018/06/28 Python
手把手教你如何安装Pycharm(详细图文教程)
2018/11/28 Python
python 将有序数组转换为二叉树的方法
2019/03/26 Python
解决python gdal投影坐标系转换的问题
2020/01/17 Python
python编程进阶之类和对象用法实例分析
2020/02/21 Python
python 安装教程之Pycharm安装及配置字体主题,换行,自动更新
2020/03/13 Python
Ralph Lauren意大利官方网站：时尚界最负盛名的品牌之一
2018/10/18 全球购物
橄榄树药房：OLIVEDA
2019/09/01 全球购物
医院实习介绍信
2014/01/12 职场文书
国家助学金获奖感言
2014/01/31 职场文书
英语专业毕业生求职信
2014/05/24 职场文书
应届生面试求职信
2014/07/02 职场文书