编程 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网络编程学习笔记(八)：XML生成与解析（DOM、ElementTree）

Jun 09 Python

python实现定时自动备份文件到其他主机的实例代码

Feb 23 Python

python实现逐个读取txt字符并修改

Dec 24 Python

pyinstaller打包多个py文件和去除cmd黑框的方法

Jun 21 Python

对django的User模型和四种扩展/重写方法小结

Aug 17 Python

pytorch载入预训练模型后,实现训练指定层

Jan 06 Python

python正则过滤字母、中文、数字及特殊字符方法详解

Feb 11 Python

jupyternotebook 撤销删除的操作方式

Apr 17 Python

通过Python pyecharts输出保存图片代码实例

Nov 25 Python

Python echarts实现数据可视化实例详解

Mar 03 Python

Python使用华为API为图像设置多个锚点标签

Apr 12 Python

Python sklearn分类决策树方法详解

Sep 23 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

2006/10/09 PHP

使用PHP遍历文件目录与清除目录中文件的实现详解

2013/06/24 PHP

如何在Ubuntu下启动Apache的Rewrite功能

2013/07/05 PHP

PHP+Redis事务解决高并发下商品超卖问题(推荐)

2020/08/03 PHP

Expandable "Detail" Table Rows

2007/08/29 Javascript

使用原生js写的一个简单slider

2014/04/29 Javascript

node.js中的console.trace方法使用说明

2014/12/09 Javascript

jQuery中:first选择器用法实例

2014/12/30 Javascript

jQuery修改class属性和CSS样式整理

2015/01/30 Javascript

jquery实现简单的全选和反选功能

2016/01/02 Javascript

jQuery中ajax的load()与post()方法实例详解

2016/01/05 Javascript

jQuery实现为LI列表前3行设置样式的方法【2种方法】

2016/09/04 Javascript

H5移动端适配 Flexible方案

2016/10/24 Javascript

微信小程序访问node.js接口服务器搭建教程

2017/04/25 Javascript

angularJS 发起$http.post和$http.get请求的实现方法

2017/05/18 Javascript

javascript获取指定区间范围随机数的方法

2017/09/08 Javascript

H5实现仿flash效果的实现代码

2017/09/29 Javascript

node.js使用免费的阿里云ip查询获取ip所在地【推荐】

2018/09/03 Javascript

Vue实现计算器计算效果

2020/08/17 Javascript

JS如何操作DOM基于表格动态展示数据

2020/10/15 Javascript

使用Vant完成Dialog弹框案例

2020/11/11 Javascript

python+opencv实现高斯平滑滤波

2020/07/21 Python

python把1变成01的步骤总结

2019/02/27 Python

Pytorch Tensor的索引与切片例子

2019/08/18 Python

浅谈对pytroch中torch.autograd.backward的思考

2019/12/27 Python

pandas数据处理之绘图的实现

2020/06/15 Python

如何在keras中添加自己的优化器(如adam等)

2020/06/19 Python

用pandas划分数据集实现训练集和测试集

2020/07/20 Python

草莓网美国官网：Strawberrynet USA

2016/12/11 全球购物

什么是SQL Server的确定性函数和不确定性函数

2016/08/04 面试题

二年级数学教学反思

2014/01/21 职场文书

尽职尽责村干部自我鉴定

2014/01/23 职场文书

医德考评自我评价

2014/09/14 职场文书

销售员态度差检讨书

2014/10/26 职场文书

WordPress多语言翻译插件 - WPML使用教程

2021/04/01 PHP

Alexa停服！网站排名将何去何从？目前还没有替代品。

2022/04/15 杂记