粗暴解决CUDA out of memory的问题
更新时间：2021年05月22日 10:35:57 作者：李橡树
这篇文章主要介绍了粗暴解决CUDA out of memory的问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
小渣渣复现大佬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
大功告成！
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
VScode查看python f.write()的文件乱码问题及解决方法
这篇文章主要介绍了VScode查看python f.write()的文件乱码问题及解决方法,本文通过图文并茂的形式给大家分享解决方法，需要的朋友可以参考下2023-02-02
这篇文章主要介绍了在Python的Django框架中包装视图函数的方法,即requires_login的相关方法,需要的朋友可以参考下2015-07-07
这篇文章主要介绍了python爬取拉勾网数据并进行数据可视化,爬取拉勾网关于python职位相关的数据信息，并将爬取的数据已csv各式存入文件，然后对csv文件相关字段的数据进行清洗，并对数据可视化展示，包括柱状图展示、直方图展示,需要的朋友可以参考下2019-07-07
本文介绍了 python 内部是如何管理字符串对象，以及字符串查找操作是如何实现的，感兴趣的小伙伴们可以参考一下2015-12-12
这篇文章主要介绍了Python编程实现蚁群算法详解，涉及蚂蚁算法的简介，主要原理及公式，以及Python中的实现代码，具有一定参考价值，需要的朋友可以了解下2021-10-10
Python面向对象程序设计类变量与成员变量、类方法与成员方法用法分析
这篇文章主要介绍了Python面向对象程序设计类变量与成员变量、类方法与成员方法用法,结合实例形式较为详细的分析了类变量与成员变量、类方法与成员方法、类方法与静态方法等概念、原理及使用技巧,需要的朋友可以参考下2019-04-04
今天给大家带来的是关于Python的相关知识,文章围绕着如何用Python实现感知器算法展开,文中有非常详细的介绍及代码示例,需要的朋友可以参考下2021-06-06
下面小编就为大家带来一篇Python按行读取文件的简单实现方法。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2016-06-06
这篇文章主要介绍了Django如何实现防止XSS攻击,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-10-10
在 Python 中,断点是一种在代码中设置的标记,用于在程序执行过程中停止或中断程序的执行,本文将详细介绍 Python 中的断点类型,感兴趣的可以了解下2024-02-02
最新评论