编程 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脚本实现分析dns日志并对受访域名排行

Sep 18 Python

python中pygame模块用法实例

Oct 09 Python

Python多线程编程（八）：使用Event实现线程间通信

Apr 05 Python

使用apidocJs快速生成在线文档的实例讲解

Feb 07 Python

python 读取txt中每行数据,并且保存到excel中的实例

Apr 29 Python

解决python 未发现数据源名称并且未指定默认驱动程序的问题

Dec 07 Python

Python 获取主机ip与hostname的方法

Dec 17 Python

PyQt5多线程刷新界面防假死示例

Dec 13 Python

Python连接Hadoop数据中遇到的各种坑(汇总)

Apr 14 Python

如何表示python中的相对路径

Jul 08 Python

Python3爬虫中Selenium的用法详解

Jul 10 Python

Python tempfile模块生成临时文件和临时目录

Sep 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

AJAX的跨域访问-两种有效的解决方法介绍

2013/06/22 PHP

ThinkPHP安装和设置

2015/07/27 PHP

使用PHPMailer发送邮件实例

2017/02/15 PHP

php使用curl实现简单模拟提交表单功能

2017/05/15 PHP

基于PHP-FPM进程池探秘

2017/10/17 PHP

PHP面向对象程序设计（OOP）之方法重写（override）操作示例

2018/12/21 PHP

php 使用mpdf实现指定字段配置字体样式的方法

2019/07/29 PHP

为开发者准备的10款最好的jQuery日历插件

2014/02/04 Javascript

JS实现图片无间断滚动代码汇总

2014/07/30 Javascript

a标签的href与onclick事件的区别详解

2014/11/12 Javascript

jQuery制作可自定义大小的拼图游戏

2015/03/30 Javascript

JS实现网页上随滚动条滚动的层效果代码

2015/11/04 Javascript

JavaScript setTimeout使用闭包功能实现定时打印数值

2015/12/18 Javascript

深入理解JQuery中的事件与动画

2016/05/18 Javascript

bootstrap的3级菜单样式,支持母版页保留打开状态实现方法

2016/11/10 Javascript

Angular中的interceptors拦截器

2017/06/25 Javascript

使用 webpack 插件自动生成 vue 路由文件的方法

2019/08/20 Javascript

vue实现Input输入框模糊查询方法

2021/01/29 Javascript

JS数组reduce()方法原理及使用技巧解析

2020/07/14 Javascript

简单介绍Python的Django框架加载模版的方式

2015/07/20 Python

Python绑定方法与非绑定方法详解

2017/08/18 Python

Python实现图片转字符画的示例

2017/08/22 Python

用Python写一个模拟qq聊天小程序的代码实例

2019/03/06 Python

Python函数装饰器原理与用法详解

2019/08/16 Python

python标准库OS模块详解

2020/03/10 Python

Python configparser模块操作代码实例

2020/06/08 Python

泰国第一在线超市：Tops

2021/02/13 全球购物

2014年情人节活动方案

2014/02/16 职场文书

教职工代表大会主持词

2014/04/01 职场文书

节约能源标语

2014/06/17 职场文书

2014年党员个人工作总结

2014/12/02 职场文书

大学生旷课检讨书1000字

2015/02/19 职场文书

超市督导岗位职责

2015/04/10 职场文书

导游词之宁夏贺兰山岩画

2019/11/08 职场文书

Django实现聊天机器人

2021/05/31 Python

tomcat的catalina.out日志按自定义时间格式进行分割的操作方法

2022/04/02 Servers