首页
所有分类
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运算π的值深入Python中科学计算的实现
Apr 17 Python
浅谈Python由__dict__和dir()引发的一些思考
Oct 30 Python
Python2.7 实现引入自己写的类方法
Apr 29 Python
对python sklearn one-hot编码详解
Jul 10 Python
Python面向对象程序设计之继承与多继承用法分析
Jul 13 Python
详解Python字典的操作
Mar 04 Python
Django框架 查询Extra功能实现解析
Sep 04 Python
Python多线程爬取豆瓣影评API接口
Oct 22 Python
python爬虫中PhantomJS加载页面的实例方法
Nov 12 Python
python re的findall和finditer的区别详解
Nov 15 Python
Anaconda的安装与虚拟环境建立
Nov 18 Python
Python将QQ聊天记录生成词云的示例代码
Feb 10 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基础之进程详解
GNN(1) SFTP(3) 文字识别(1) 追加文本(1) GCN(1) 验证码(1) 柱状图(1) Pytesseract(1) 匹配文本(1) flex-shrink(1)
You might like
mysql总结之explain
2012/02/27 PHP
PHP文件缓存类示例分享
2015/01/30 PHP
PHP CURL采集百度搜寻结果图片不显示问题的解决方法
2017/02/03 PHP
PHP使用Redis实现防止大并发下二次写入的方法
2017/10/09 PHP
Laravel-admin之修改操作日志的方法
2019/09/30 PHP
juqery 学习之三 选择器 简单 内容
2010/11/25 Javascript
Colortip基于jquery的信息提示框插件在IE6下面的显示问题修正方法
2010/12/06 Javascript
javascript实现文字图片上下滚动的具体实例
2013/06/28 Javascript
js判断字符长度及中英文数字等
2014/03/19 Javascript
javascript实现抽奖程序的简单实例
2016/06/07 Javascript
Javascript之面向对象--封装
2016/12/02 Javascript
EasyUI框架 使用Ajax提交注册信息的实现代码
2017/09/27 Javascript
JavaScript 保护变量不被随意修改的实现代码
2017/09/27 Javascript
AngularJS对动态增加的DOM实现ng-keyup事件示例
2018/03/12 Javascript
JS中this的指向以及call、apply的作用
2018/05/06 Javascript
JS函数动态传递参数的方法分析【基于arguments对象】
2019/06/05 Javascript
Vue computed 计算属性代码实例
2020/04/22 Javascript
[07:26]2015国际邀请赛第二日TOP10集锦
2015/08/06 DOTA
比较详细Python正则表达式操作指南(re使用)
2008/09/06 Python
利用Python抓取行政区划码的方法
2016/11/28 Python
Tensorflow 自带可视化Tensorboard使用方法(附项目代码)
2018/02/10 Python
在python中,使用scatter绘制散点图的实例
2019/07/03 Python
opencv 获取rtsp流媒体视频的实现方法
2019/08/23 Python
Numpy与Pytorch 矩阵操作方式
2019/12/27 Python
Pytorch 多维数组运算过程的索引处理方式
2019/12/27 Python
在pycharm中使用matplotlib.pyplot 绘图时报错的解决
2020/06/01 Python
详解python metaclass（元类）
2020/08/13 Python
Python私有属性私有方法应用实例解析
2020/09/15 Python
Bluebella美国官网：英国性感内衣品牌
2018/10/04 全球购物
德国受欢迎的旅游和休闲网站：lastminute.de
2019/09/23 全球购物
倡议书格式
2014/04/14 职场文书
医院标语大全
2014/06/23 职场文书
党支部创先争优活动总结
2014/08/28 职场文书
Python如何解决secure_filename对中文不支持问题
2021/07/16 Python
MySQL之MyISAM存储引擎的非聚簇索引详解
2022/03/03 MySQL
python实现会员管理系统
2022/03/18 Python