粗暴解决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
大功告成！
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python部署web app的详细教程,示例代码基于Python2.x版本,需要的朋友可以参考下2015-04-04
Python3.4 splinter(模拟填写表单)使用方法
今天小编就为大家分享一篇Python3.4 splinter(模拟填写表单)使用方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-10-10
这篇文章主要介绍了如何在Python中引用其他模块，在Python中，除了可以自定义模块外，还可以引用其他模块，主要包括使用标准库和第三方模块，下面文章分别详细的介绍，需要的小伙伴可以参考一下2022-06-06
这篇文章主要介绍了Python使用Matplotlib绘制3D曲面图,在Python中,使用Matplotlib库绘制3D曲面图可以通过mpl_toolkits.mplot3d模块实现,需要的朋友可以参考下2025-05-05
在数据驱动的时代,Excel仍是大量企业存储核心数据的工具,Python的Pandas库凭借其向量化计算、内存优化和丰富的数据处理接口,成为自动化分析Excel数据的首选工具,下面我们就来看看具体应用吧2025-05-05
这篇文章主要介绍了python高效的素数判断算法，研究算法的同学一定要看一下2021-04-04
这篇文章主要介绍了Pytorch BertModel的使用说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
这篇文章主要介绍了如何基于Python中ArcPy模块，对大量不同时相的栅格遥感影像按照其成像时间依次执行批量拼接的方法，感兴趣的可以了解一下2023-03-03
Python实例一个类背后发生了什么，本文为大家一一列出，感兴趣的朋友可以参考一下2016-02-02
python中sklearn库predict及python sklearn linearmodel(
Scikit-learn(sklearn)是机器学习中常用的第三方模块,对常用的机器学习方法进行了封装,包括回归(Regression)、降维(Dimensionality Reduction)、分类(Classfication)、聚类(Clustering)等方法,今天小编给大家分享python中sklearn库predict的问题,感兴趣的朋友一起看看吧2024-02-02
最新评论