粗暴解决CUDA out of memory的问题
作者：李橡树 发布时间：2023-05-12 05:40:00
标签：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
大功告成！
来源：https://blog.csdn.net/weixin_42507051/article/details/90757508
0
投稿猜你喜欢
- 一、查看實例名時可用1、服务—SQL Server(实例名)，默认实例为(MSSQLSERVER)或在连接企业管理时-查看本地实例2、通過注
- Python的装饰器可以实现在代码运行期间修改函数的上下文， 即可以定义函数在执行之前进行何种操作和函数执行后进行何种操作， 而函数本身并没
- 假设三节点MGR某个节点异常，需要重新把这个节点加入到MGR集群中，具体操作过程如下：贡献者端执行（192.168.1.11）DROP US
本文实例讲述了JS实现获取数组中最大值或最小值功能。分享给大家供大家参考，具体如下：方法一：//最小值Array.prototype.min- 本文实例为大家分享了python实现局域网内聊天功能的具体代码，供大家参考，具体内容如下功能: 可以向局域网内开启接收信息功能的ip进行发送
- python窃取摄像头照片源码+获取授权码方法+py打包成exe教你用python做一个属于自己的窃取摄像头照片的软件。需要安装python
- 如果用树作为索引的数据结构，每查找一次数据就会从磁盘中读取树的一个节点，也就是一页，而二叉树的每个节点只存储一条数据，并不能填满一页的存储空
- 本文实例为大家分享了python批量文件重命名的具体代码，供大家参考，具体内容如下问题描述最近遇到朋友求助，如何将大量文件名前面的某些字符删
- sqlx是Golang中的一个知名三方库，其为Go标准库database/sql提供了一组扩展支持。使用它可以方便的在数据行与Golang的
- 微信的小程序是一个很不错的体验，简单，上手快，这几天也在学习使用小程序，自己总结了三种用 Python 作为小程序后端的方式，供你参考。方法
- 本文实例讲述了php+html5基于websocket实现聊天室的方法。分享给大家供大家参考。具体如下：html5的websocket 实现
- 大家是否还记得1983年任天堂的著名游戏《超级玛丽》里那个留着胡子的意大利水管工人，还有日本konami公司1987年发行的射击游戏《魂斗罗
- 在进行深度学习实验时，GPU 的实时状态监测十分有必要。今天详细解读一下 nvidia-smi 命令上图是服务器上 GeForce GTX
- python中使用input()函数来获取用户输入函数 input() 让程序暂停运行，等待用户输入一些文本，获取用户的输入后，Python
- 这篇文章主要介绍了python plotly画柱状图代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需
- 一、方法2此方法是两个表构建某一相同字段，然后全连接，在做匹配结果筛选，此方法针对数据量不大的时候，逻辑比较简单，但是内存消耗较大1. 导入
- import timedef block(file,size=65536): while True:&n
- 联合索引又叫复合索引。对于复合索引:Mysql从左到右的使用索引中的字段，一个查询可以只使用索引中的一部份，但只能是最左侧部分。例如索引是k
- 当我们拿到一个对象的引用时，如何知道这个对象是什么类型、有哪些方法呢？使用type()首先，我们来判断对象类型，使用type()函数：基本类
- 今天做了一个很简单的小项目，感受到了paramiko模块的强大，也深感自己Linux的功力不行~~一、需求二、简单需求分析及流程图需求很少，
