粗暴解决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中的urllib模块使用详解
Jul 07 Python
python去除空格和换行符的实现方法(推荐)
Jan 04 Python
Python及PyCharm下载与安装教程
Nov 18 Python
Python使用requests发送POST请求实例代码
Jan 25 Python
python操作oracle的完整教程分享
Jan 30 Python
Python Web程序部署到Ubuntu服务器上的方法
Feb 22 Python
关于django 数据库迁移(migrate)应该知道的一些事
May 27 Python
Django中使用 Closure Table 储存无限分级数据
Jun 06 Python
Python 使用 PyMysql、DBUtils 创建连接池提升性能
Aug 14 Python
python批量替换文件名中的共同字符实例
Mar 05 Python
Python如何使用神经网络进行简单文本分类
Feb 25 Python
python实现监听键盘
Apr 26 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
CI框架给视图添加动态数据
2014/12/01 PHP
PHP向socket服务器收发数据的方法
2015/01/24 PHP
基础的WordPress插件制作教程
2015/11/24 PHP
PHP实现简单实用的分页类代码
2016/04/08 PHP
php 微信开发获取用户信息如何实现
2016/12/13 PHP
php获取手机端的号码以及ip地址实例代码
2018/09/12 PHP
网易JS面试题与Javascript词法作用域说明
2010/11/09 Javascript
在javascript中对于DOM的加强
2013/04/11 Javascript
Jquery操作radio的简单实例
2014/01/06 Javascript
JavaScript实现的购物车效果可以运用在好多地方
2014/05/09 Javascript
javascript内置对象操作详解
2015/02/04 Javascript
jquery实现两个图片渐变切换效果的方法
2015/06/25 Javascript
基于Jquery easyui 选中特定的tab
2015/11/17 Javascript
详解堆的javascript实现方法
2016/11/29 Javascript
jQuery实现百度登录框的动态切换效果
2017/04/21 jQuery
解决element-ui里的下拉多选框 el-select 时,默认值不可删除问题
2020/08/14 Javascript
零基础学Python(一)Python环境安装
2014/08/20 Python
Python实现计算文件夹下.h和.cpp文件的总行数
2015/04/23 Python
详解Python之unittest单元测试代码
2018/01/24 Python
python爬虫爬取快手视频多线程下载功能
2018/02/28 Python
Python模块future用法原理详解
2020/01/20 Python
PyQt5高级界面控件之QTableWidget的具体使用方法
2020/02/23 Python
pytorch VGG11识别cifar10数据集(训练+预测单张输入图片操作)
2020/06/24 Python
详解background属性的8个属性值(面试题)
2020/11/02 HTML / CSS
HTML5里autofocus自动聚焦属性使用介绍
2016/06/22 HTML / CSS
简述安装Slackware Linux系统的过程
2012/01/12 面试题
电信专业毕业生推荐信
2013/11/18 职场文书
《鲁班和橹板》教学反思
2014/04/27 职场文书
篮球友谊赛通讯稿
2014/10/10 职场文书
工作批评与自我批评范文
2014/10/16 职场文书
2014年转正工作总结
2014/11/08 职场文书
优秀党支部申报材料
2014/12/24 职场文书
搞笑婚前保证书
2015/02/28 职场文书
学校远程教育工作总结
2015/08/11 职场文书
2016年中学法制宣传日活动总结
2016/04/01 职场文书
MIME类型中application/xml与text/xml的区别介绍
2022/01/18 HTML / CSS