RuntimeError: CUDA out of memory

最新推荐文章于 2024-06-14 16:03:35 发布
最新推荐文章于 2024-06-14 16:03:35 发布
阅读量256 收藏 1
点赞数 1
文章标签： 机器学习 深度学习 神经网络 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/YXYXYX12/article/details/111171225
版权

在ST-GCN中，train model时出现的错误RuntimeError: CUDA out of memory

  1. 出现的具体问题如下图所示，在开始训练模型的第4个epoch跳出了RuntimeError的错误。在尝试使用少量的数据集（752MB）以及改小batch_size的前提下，仍然再第4个epoch终止后开始计算平均loss时，出现错误。苦思不得其解，于是遍览st-gcn的github的Issues，看到了这一个解决方法：

在这里插入图片描述
在“python main.py recognition -c config/st_gcn/ntu-xsub/train.yaml”此命令行后加上对test_bacth_size的限定，预留出存储空间，我也是抱着尝试的心态把命令行改为了“python main.py recognition -c config/st_gcn/ntu-xsub/train.yaml --test_batch_size 8”，结果证明该方法可行，如下图：
在这里插入图片描述

Thomas_yx
关注
RuntimeError: CUDA out of memory.
网易搬砖选手
04-10 1万+
我认为对于 GPU 内存较低的 PyTorch 用户来说，这是一个非常常见的消息：X MiB free;X cached)但似乎效果不是很大。虽然为清除占用的cuda内存提供了一个很好的选择，我们还可以通过使用手动清除未使用的变量，但在使用这些命令后，错误可能会再次出现，因为 pytorch 实际上并没有清除内存，而是清除了对变量占用的内存的引用。因此，重新启动内核后减小batch_size并找到最佳batch_size是最好的选择（但有时不是一个非常可行的选择）。其中，两个参数都是可选的。
RuntimeError: CUDA out of memory. Tried to allocate 1.77 GiB？如何解决
**My Coding Family**
05-27 996
ok，以上就是我这期的Bug修复内容啦，如果还想查找更多解决方案，你可以看看我专门收集Bug及提供解决方案的专栏「Bug调优」，都是实战中碰到的Bug，希望对你有所帮助。到此，咱们下期拜拜。码字不易，如果这篇文章对你有所帮助，帮忙给bugj菌来个，您的支持就是我坚持写作分享知识点传播技术的最大动力。「猿圈奇妙屋」；
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码： with torch.no_grad(): # 停止自动反向计算梯度 参考： https://wisdomai.xyz/tool/pytorch/archives/2301 https://ptorch.com/news/160.html 作者：菜叶儿掉啦
解决：RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0； 2.00 GiB total capacity； 1
地中海の养成记
01-31 6万+
1. 问题2. 分析3. 解决 1. 问题 训练模型时报错： RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 2.00 GiB total capacity; 1.49 GiB already allocated; 57.03 MiB free; 6.95 MiB cached) 2. 分析 这种问题，是GPU内存不够引起的 3. 解决 方法一： 换高性能高显存的显卡 方法二：修改代码 报错的训练代码为.
RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0； 2.00 GiB total capacity； 1.29
weixin_43725886的博客
07-04 1万+
RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0; 2.00 GiB total capacity; 1.29 GiB already allocated; 79.00 MiB free; 1.30 GiB reserved in total by PyTorch) 解决方案
RuntimeError: CUDA out of memory（已解决）
热门推荐
雷恩Layne
09-01 12万+
今天用pytorch训练神经网络时，出现如下错误： RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0; 2.00 GiB total capacity; 1.29 GiB already allocated; 79.00 MiB free; 1.30 GiB reserved in total by PyTorch) 明明 GPU 0 有2G容量，为什么只有 79M 可用？ 并且 1.30G已经被PyTorch占用了。
AI绘画——使用stable-diffusion生成图片时提示RuntimeError: CUDA out of memory处理方法
PromiseTo的专栏
10-10 2万+
以半精度（fp16）使用该模型，因为它几乎总是提供与全精度相同的结果，但速度大约是其两倍，并且需要一半的GPU RAM量。修改scripts/txt2img.py文件，在。，一次生成一张图片，进一步降低显存。
RuntimeError:CUDA error:out of memory
sinat_37024730的博客
11-02 192
RuntimeError:CUDA error:out of memory // cmd》nvidia-smi 查看gpu-memory现在的状态 import os os.environ["CUDA_VISIBLE_DEVICES"] = '1'
RuntimeError: CUDA error: out of memory
⚡
03-07 381
执行torch.load导入某个.pt。
PyTorch训练模型时RuntimeError: CUDA out of memory. Tried to allocate 2.24 GiB (GPU 0； 1.83 GiB total capa
二爷的博客
03-26 571
【代码】PyTorch训练模型时RuntimeError: CUDA out of memory. Tried to allocate 2.24 GiB (GPU 0;
测试时报错 RuntimeError: CUDA out of memory.
齐天大胜
12-19 3232
问题描述： 训练神经网络模型，训练时正常，训练一个epoch后测试的时候，报错RuntimeError: CUDA out of memory. 然而我的模型并没有很大，GPU占用率也不高。 解决方法： with torch.no_grad(): (your test code) ...
RuntimeError: CUDA out of memory. 错误日志
A_Tu_daddy的博客
04-08 1240
运行代码时出现的错误如下： RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 4.00 GiB total capacity; 2.91 GiB already allocated; 166.40 KiB free; 2.93 GiB reserved in total by PyTorch) 看了一下自己...
错误合集1：RuntimeError: CUDA out of memory. Tried to allocate 196.00 MiB (GPU 0； 23.65 GiB total capacit
最新发布
m0_53587476的博客
06-14 542
RuntimeError: CUDA out of memory. Tried to allocate 196.00 MiB (GPU 0; 23.65 GiB total capacity; 21.34 GiB already allocated; 48.06 MiB free; 21.35 GiB reserved in total by PyTorch) If reserved memory问题。
解决：RuntimeError: CUDA out of memory. Tried to allocate 160.00 MiB (GPU 0； 10.76 GiB total capacity..
zcyzcyjava的博客
10-25 2万+
内存分配不足：需要160MB，，但GPU只剩下135.31MB，
模型推理runtimeerror: cuda out of memory
09-17
模型推理中出现"RuntimeError: CUDA out of memory"错误是因为在GPU上使用的显存不足。这个错误通常发生在模型的输入数据量过大或模型占用的显存资源过多时。 解决这个问题可以采用以下方法： 1. 减少批处理大小：...
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值