编程 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 RPC服务器端和客户端实例

Nov 22 Python

Python实现分割文件及合并文件的方法

Jul 10 Python

Python编程中的异常处理教程

Aug 21 Python

Python编程实现的简单神经网络算法示例

Jan 26 Python

Python3.7中安装openCV库的方法

Jul 11 Python

详解Python中正则匹配TAB及空格的小技巧

Jul 26 Python

有关Tensorflow梯度下降常用的优化方法分享

Feb 04 Python

Pytorch中.new()的作用详解

Feb 18 Python

Django 设置多环境配置文件载入问题

Feb 25 Python

详细分析Python collections工具库

Jul 16 Python

Python基础之变量的相关知识总结

Jun 23 Python

python前后端自定义分页器

Apr 13 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

索尼ICF-SW100收音机评测

2021/03/02 无线电

php 3行代码的分页算法(求起始页和结束页)

2009/10/21 PHP

Codeigniter中mkdir创建目录遇到权限问题和解决方法

2014/07/25 PHP

twig模板获取全局变量的方法

2016/02/05 PHP

PHP 断点续传实例详解

2017/11/11 PHP

dwr spring的集成实现代码

2009/03/22 Javascript

重载toString实现JS HashMap分析

2011/03/13 Javascript

html中table数据排序的js代码

2011/08/09 Javascript

Javascript alert消息换行的方法

2013/08/07 Javascript

JavaScript fontsize方法入门实例（按照指定的尺寸来显示字符串）

2014/10/17 Javascript

AngularJs concepts详解及示例代码

2016/09/01 Javascript

NodeJS使用formidable实现文件上传

2016/10/27 NodeJs

javascript 利用arguments实现可变长参数

2016/11/21 Javascript

详解webpack-dev-server 设置反向代理解决跨域问题

2018/04/18 Javascript

vue动态合并单元格并添加小计合计功能示例

2020/11/26 Vue.js

python Django模板的使用方法

2016/01/14 Python

python socket多线程通讯实例分析(聊天室)

2016/04/06 Python

Python实现TCP/IP协议下的端口转发及重定向示例

2016/06/14 Python

Django教程笔记之中间件middleware详解

2018/08/01 Python

使用python将最新的测试报告以附件的形式发到指定邮箱

2019/09/20 Python

Python-jenkins模块获取jobs的执行状态操作

2020/05/12 Python

python 如何实现遗传算法

2020/09/22 Python

Selenium执行完毕未关闭chromedriver/geckodriver进程的解决办法(java版+python版)

2020/12/07 Python

Marks & Spencer爱尔兰：英国马莎百货

2016/04/20 全球购物

Expedia爱尔兰：酒店、机票、租车及廉价假期

2017/01/02 全球购物

Staples英国官方网站：办公用品一站式采购

2017/10/06 全球购物

捷克汽车配件和工具销售网站：TorriaCars

2018/02/26 全球购物

英国领先的品牌珠宝和配件供应商：Acotis Jewellery

2018/03/07 全球购物

全球性的奢侈品梦工厂：Forzieri（福喜利）

2019/02/20 全球购物

实习生个人的自我评价

2013/12/08 职场文书

求职信范文英文版

2014/01/05 职场文书

党的群众路线教育实践活动批评与自我批评

2014/02/16 职场文书

2014年大学生四年规划书范文

2014/04/03 职场文书

解除劳动合同协议书

2014/04/14 职场文书

考试作弊检讨书

2014/10/21 职场文书

浅谈Web Storage API的使用

2021/06/23 Javascript