编程 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网络编程学习笔记(九)：数据库客户端 DB-API

Jun 09 Python

Python datetime时间格式化去掉前导0

Jul 31 Python

python删除列表内容

Aug 04 Python

Python在信息学竞赛中的运用及Python的基本用法(详解)

Aug 15 Python

win10下Python3.6安装、配置以及pip安装包教程

Oct 01 Python

Python实现动态图解析、合成与倒放

Jan 18 Python

python中的插值 scipy-interp的实现代码

Jul 23 Python

django项目简单调取百度翻译接口的方法

Aug 06 Python

浅谈python3 构造函数和析构函数

Mar 12 Python

python编写一个会算账的脚本的示例代码

Jun 02 Python

Python实现Canny及Hough算法代码实例解析

Aug 06 Python

全网最详细的PyCharm+Anaconda的安装过程图解

Jan 25 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

AM/FM收音机的安装与调试

2021/03/02 无线电

PHP编码转换函数自动转换字符集支持数组转换

2012/12/16 PHP

php从字符串创建函数的方法

2015/03/16 PHP

实现PHP框架系列文章(6)mysql数据库方法

2016/03/04 PHP

完美解决phpdoc导出文档中@package的warning及Error的错误

2016/05/17 PHP

PHP函数积累总结

2019/03/19 PHP

使用git迁移Laravel项目至新开发环境的步骤详解

2020/04/06 PHP

JavaScript 联动的无限级封装类,数据采用非Ajax方式,随意添加联动

2010/06/29 Javascript

JQuery小知识

2010/10/15 Javascript

JavaScript学习笔记（一） js基本语法

2011/10/25 Javascript

jquery实现table鼠标经过变色代码

2013/09/25 Javascript

js获取窗口相对于屏幕左边和上边的位置坐标

2014/05/15 Javascript

JS获取鼠标选中的文字

2016/08/10 Javascript

微信小程序开发之实现自定义Toast弹框

2017/06/08 Javascript

JS自定义滚动条效果简单实现代码

2020/10/27 Javascript

JavaScript模块详解

2017/12/18 Javascript

微信小程序使用for循环动态渲染页面操作示例

2018/12/25 Javascript

微信小程序登录对接Django后端实现JWT方式验证登录详解

2019/07/29 Javascript

vue.js实现左边导航切换右边内容

2019/10/21 Javascript

JavaScript定时器常见用法实例分析

2019/11/15 Javascript

python中 ? : 三元表达式的使用介绍

2013/10/09 Python

利用Python中的mock库对Python代码进行模拟测试

2015/04/16 Python

python实现二叉树的遍历

2017/12/11 Python

numpy.linspace 生成等差数组的方法

2018/07/02 Python

Django集成celery发送异步邮件实例

2019/12/17 Python

python+adb+monkey实现Rom稳定性测试详解

2020/04/23 Python

利用python+request通过接口实现人员通行记录上传功能

2021/01/13 Python

Canvas环形饼图与手势控制的实现代码

2019/11/08 HTML / CSS

canvas绘制图片drawImage使用方法

2020/09/15 HTML / CSS

Carter’s官方旗舰店：美国受欢迎的婴童服装品牌

2018/01/21 全球购物

美国帽子俱乐部商店：Hat Club

2019/07/05 全球购物

施工安全标语

2014/06/07 职场文书

2014最新股权信托合同协议书

2014/11/18 职场文书

公司欠款证明

2015/06/24 职场文书

springboot 多数据源配置不生效遇到的坑及解决

2021/11/17 Java/Android

微信告警的zabbix监控系统监控整个NGINX集群

2022/04/18 Servers