编程 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检测手机QQ在线状态的脚本代码

Feb 10 Python

用Python制作检测Linux运行信息的工具的教程

Apr 01 Python

几个提升Python运行效率的方法之间的对比

Apr 03 Python

Python中用于计算对数的log()方法

May 15 Python

Python打印斐波拉契数列实例

Jul 07 Python

Python模拟百度登录实例详解

Jan 20 Python

利用python将图片转换成excel文档格式

Dec 30 Python

Python面向对象编程之继承与多态详解

Jan 16 Python

如何修复使用 Python ORM 工具 SQLAlchemy 时的常见陷阱

Nov 19 Python

python实现高斯投影正反算方式

Jan 17 Python

python基于OpenCV模板匹配识别图片中的数字

Mar 31 Python

Python音乐爬虫完美绕过反爬

Aug 30 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

模拟SQLSERVER的两个函数：dateadd(),datediff()

2006/10/09 PHP

php下过滤html代码的函数提高程序安全性

2010/03/02 PHP

优化php效率,提高php性能的一些方法

2011/03/24 PHP

解析isset与is_null的区别

2013/08/09 PHP

PHP计算指定日期所在周的开始和结束日期的方法

2015/03/24 PHP

JSON用法之将PHP数组转JS数组，JS如何接收PHP数组

2015/10/08 PHP

PHP实现二叉树深度优先遍历(前序、中序、后序)和广度优先遍历(层次)实例详解

2018/04/20 PHP

Javascript 兼容firefox的一些问题

2009/05/21 Javascript

Javascript alert消息换行的方法

2013/08/07 Javascript

javascript的propertyIsEnumerable()方法使用介绍

2014/04/09 Javascript

js跨域访问示例(客户端/服务端)

2014/05/19 Javascript

js监听鼠标点击和键盘点击事件并自动跳转页面

2014/09/24 Javascript

基于jQuery实现复选框的全选全不选反选功能

2014/11/24 Javascript

HTML5+setCutomValidity()函数验证表单实例分享

2015/04/24 Javascript

基于BootStrap Metronic开发框架经验小结【三】下拉列表Select2插件的使用

2016/05/12 Javascript

jQuery实现的无限级下拉菜单功能示例

2016/09/12 Javascript

VUE JS 使用组件实现双向绑定的示例代码

2017/01/10 Javascript

详解微信小程序入门五: wxml文件引用、模版、生命周期

2017/01/20 Javascript

详解原生js实现offset方法

2017/06/15 Javascript

利用jquery去掉时光轴头尾部线条的方法实例

2017/06/16 jQuery

js+html获取系统当前时间

2017/11/10 Javascript

解决低版本的浏览器不支持es6的import问题

2018/03/09 Javascript

使用 Node.js 实现图片的动态裁切及算法实例代码详解

2018/09/29 Javascript

vue图片加载失败时用默认图片替换的方法

2019/08/29 Javascript

JQuery使用数组遍历跳出each循环

2020/09/01 jQuery

微信小程序实现单个或多个倒计时功能

2020/11/01 Javascript

vue添加自定义右键菜单的完整实例

2020/12/08 Vue.js

详解vue-cli项目在IE浏览器打开报错解决方法

2020/12/10 Vue.js

[56:35]DOTA2上海特级锦标赛主赛事日 - 5 总决赛Liquid VS Secret第一局

2016/03/06 DOTA

python 多线程应用介绍

2012/12/19 Python

python比较2个xml内容的方法

2015/05/11 Python

对pandas写入读取h5文件的方法详解

2018/12/28 Python

python三方库之requests的快速上手

2019/03/04 Python

Python绘图实现台风路径可视化代码实例

2020/10/23 Python

python中pyqtgraph知识点总结

2021/01/26 Python

nginx日志格式分析和修改

2022/04/28 Servers