编程 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实现模拟按键，自动翻页看u17漫画

Mar 17 Python

详解Python的迭代器、生成器以及相关的itertools包

Apr 02 Python

Python的函数的一些高阶特性

Apr 27 Python

Python简单生成随机姓名的方法示例

Dec 27 Python

python实现数据库跨服务器迁移

Apr 12 Python

python实现音乐下载器

Apr 15 Python

网红编程语言Python将纳入高考你怎么看?

Jun 07 Python

Python实现判断一个整数是否为回文数算法示例

Mar 02 Python

Django使用模板后无法找到静态资源文件问题解决

Jul 19 Python

Python学习工具jupyter notebook安装及用法解析

Oct 23 Python

5 分钟读懂Python 中的 Hook 钩子函数

Dec 09 Python

为了顺利买到演唱会的票用Python制作了自动抢票的脚本

Oct 16 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

php 多线程上下文中安全写文件实现代码

2009/12/28 PHP

php中使用Curl、socket、file_get_contents三种方法POST提交数据

2011/08/12 PHP

ThinkPHP的URL重写问题

2014/06/22 PHP

php使用curl_init()和curl_multi_init()多线程的速度比较详解

2018/08/15 PHP

jQuery 1.5最新版本的改进细节分析

2011/01/19 Javascript

关于二级域名下使用一级域名下的COOKIE的问题

2011/11/07 Javascript

jQuery 全选/反选以及单击行改变背景色实例

2013/07/02 Javascript

JS动态修改图片的URL（src）的方法

2015/04/01 Javascript

JS中Location使用详解

2015/05/12 Javascript

JS+CSS实现DIV层的展开、收缩效果

2016/01/28 Javascript

JavaScript正则获取地址栏中参数的方法

2017/03/02 Javascript

JS完成画圆圈的小球

2017/03/07 Javascript

微信小程序自定义Toast实例代码

2017/06/12 Javascript

vue.js数据绑定的方法（单向、双向和一次性绑定）

2017/07/13 Javascript

使用Vue动态生成form表单的实例代码

2018/04/26 Javascript

JQuery常用选择器功能与用法实例分析

2019/12/23 jQuery

javascript实现前端分页效果

2020/06/24 Javascript

JavaScript对象访问器Getter及Setter原理解析

2020/12/08 Javascript

vue 实现click同时传入事件对象和自定义参数

2021/01/29 Vue.js

极简的Python入门指引

2015/04/01 Python

详解Numpy数组转置的三种方法T、transpose、swapaxes

2019/05/27 Python

python实现beta分布概率密度函数的方法

2019/07/08 Python

vue学习笔记之动态组件和v-once指令简单示例

2020/02/29 Python

python中复数的共轭复数知识点总结

2020/12/06 Python

利用CSS的Sass预处理器（框架）来制作居中效果

2016/03/10 HTML / CSS

为您的家、后院、车库等在线购物：Spreetail

2019/06/17 全球购物

创业计划书模版

2014/02/05 职场文书

《孔子游春》教学反思

2014/02/25 职场文书

抗震救灾标语

2014/06/26 职场文书

供电工程专业求职信

2014/08/09 职场文书

竞聘演讲稿怎么写

2014/08/28 职场文书

服务员态度差检讨书

2014/10/28 职场文书

2015年大学班主任工作总结

2015/04/30 职场文书

歌咏比赛口号大全

2015/12/25 职场文书

Python获取百度热搜的完整代码

2021/04/07 Python

go xorm框架的使用

2021/05/22 Golang