编程 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写入并获取剪切板内容的实例

May 31 Python

Python3实现的爬虫爬取数据并存入mysql数据库操作示例

Jun 06 Python

Linux 修改Python命令的方法示例

Dec 03 Python

python使用zip将list转为json的方法

Dec 31 Python

python numpy之np.random的随机数函数使用介绍

Oct 06 Python

win7下 python3.6 安装opencv 和 opencv-contrib-python解决 cv2.xfeatures2d.SIFT_create() 的问题

Oct 24 Python

使用 Python 写一个简易的抽奖程序

Dec 08 Python

python数据库开发之MongoDB安装及Python3操作MongoDB数据库详细方法与实例

Mar 18 Python

python如何求100以内的素数

May 27 Python

详解KMP算法以及python如何实现

Sep 18 Python

python利用platform模块获取系统信息

Oct 09 Python

python 闭包函数详细介绍

Apr 19 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连mysql和oracle数据库性能比较

2006/10/09 PHP

PHP内核介绍及扩展开发指南―基础知识

2011/09/11 PHP

克隆一个新项目的快捷方式

2013/04/10 PHP

PHP面向对象之领域模型+数据映射器实例(分析)

2017/06/21 PHP

PHP实现用session来实现记录用户登陆信息

2018/10/15 PHP

070823更新的一个[消息提示框]组件兼容ie7

2007/08/29 Javascript

js如何获取object类型里的键值

2014/02/18 Javascript

Javascript 高阶函数使用介绍

2015/06/15 Javascript

快速学习JavaScript的6个思维技巧

2015/10/13 Javascript

javascript中select下拉框的用法总结

2016/01/07 Javascript

当jquery ajax遇上401请求的解决方法

2016/05/19 Javascript

关于不同页面之间实现参数传递的几种方式讨论

2017/02/13 Javascript

jQuery鼠标移动图片上实现放大效果

2017/06/25 jQuery

Vue写一个简单的倒计时按钮功能

2018/04/20 Javascript

Vue框架TypeScript装饰器使用指南小结

2019/02/18 Javascript

Taro UI框架开发小程序实现左滑喜欢右滑不喜欢效果的示例代码

2020/05/18 Javascript

JavaScript数组排序的六种常见算法总结

2020/08/18 Javascript

利用 JavaScript 实现并发控制的示例代码

2020/12/31 Javascript

Python help()函数用法详解

2014/03/11 Python

在Python的Django框架中更新数据库数据的方法

2015/07/17 Python

Python2.7下安装Scrapy框架步骤教程

2017/12/22 Python

详解python中的Turtle函数库

2018/11/19 Python

Python使用pydub库对mp3与wav格式进行互转的方法

2019/01/10 Python

python批量图片处理简单示例

2019/08/06 Python

python dataframe NaN处理方式

2019/12/26 Python

python读取多层嵌套文件夹中的文件实例

2020/02/27 Python

Python实现自动打开电脑应用的示例代码

2020/04/17 Python

HTML5上传文件显示进度的实现代码

2012/08/30 HTML / CSS

HTML5离线应用与客户端存储的实现

2018/05/03 HTML / CSS

TecoBuy澳大利亚：在线电子和小工具商店

2020/06/25 全球购物

生物技术专业研究生自荐信

2013/09/22 职场文书

应聘编辑自荐信范文

2014/03/12 职场文书

员工三分钟演讲稿

2014/08/19 职场文书

小学生纪念九一八事变演讲稿

2014/09/14 职场文书

计划生育证明格式及范本

2014/10/09 职场文书

新手初学Java网络编程

2021/07/07 Java/Android