深度学习-pytorch-CUDA out of memory.错误解决方案

最新推荐文章于 2024-05-24 12:32:34 发布
最新推荐文章于 2024-05-24 12:32:34 发布
阅读量3.2k 收藏 2
点赞数 1
分类专栏： DeepLearning 文章标签： 深度学习 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_34637672/article/details/119654391
版权

最近训练模型的时候真是气死我了，花式 out of memory，明明模型不是很大，明明一个batch只读入一条音频（PS我是做语音的）。这里记录一下每一次的解决方案。

  1. 第一次溢出是因为在验证过程中，没有加不做梯度更新的设置。只要在验证之前加一个 torch.no_grad()就好了，毕竟验证不需要梯度更新。具体用法自己百度，你已经是个成熟的程序猿了，需要自己学会百度了。
  2. 第二次还是在验证的过程中，tensor计算的时候溢出了，调试的时候发现是因为tensor计算的时候维度不匹配，并没有报计算错误，而是报了CUDA out of memory。这种情况注意一下tensor之间计算时的维度要一致就可以了。
  3. …
烫烫烫烫烫火锅
关注
  • 1
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码：...
深度学习-PyTorch框架实战系列视频教程
06-28
深度学习-PyTorch框架实战系列视频教程，课程内容全部以实战为导向，基于当下计算机视觉与自然语言处理中经典项目进行实例讲解，通过Debug模式详解项目中每一行代码的作用与效果，整体风格通俗易懂，提供全部课程...
torch.cuda.OutOfMemoryError: CUDA out of memory.
热门推荐
Hello_World1023的博客
04-25 4万+
发现gpu:0显存被PID:19409程序大量占用, 报错应该是默认在gpu:0训练导致显存不足, 接着查看gpu:0上程序所属用户(如果不是师兄的我就kill了)设置完成后查看显存占用情况可以看到, gpu:1显存占用马上上升了, 不影响其他gpu卡的显存。这样即可指定在gpu:1上训练, 实际上是只设置gpu:1可见, 而屏蔽其他gpu卡。这样即可指定在gpu:1上训练, 实际上是只设置gpu:1可见, 而屏蔽其他gpu卡。发现gpu:1空闲, 指定gpu:1上训练模型, 有多种方法,
pytorch遇见RuntimeError: CUDA out of memory的解决
橙子Blue不加冰
07-26 7885
RuntimeError: CUDA out of memory 1.查看是否其他程序占用显存 遇到此类错误后，对于py格式的文件来说，程序会进行终止，也就是当前程序占用的显存将会被释放。此时可用 watch -n 1 nvidia-smi 命令查看当前显存的使用情况。如果此时显存依然有比较大的占用，说明存在其他程序占用显存，使用kill命令结束不必要的程序即可。 2.查看pytorch和cuda是否匹配 使用方法torch.cuda.is_available()，确认pytorch和cuda是否匹配，如
【已解决】torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 250.00 MiB.
最新发布
dont worry about it的博客
05-24 4万+
报错：torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 250.00 MiB.
报错记录torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 90.00 MiB (GPU 0； 7.93 GiB to
gongkeguo的博客
11-03 2万+
python报错记录
已解决：RuntimeError: CUDA error: out of memory
hh5187的博客
09-26 2万+
1. 检查是否使用了合适版本的torch print(torch.__version__) # 1.9.1+cu111 print(torch.version.cuda) # 11.1 print(torch.backends.cudnn.version()) # 8005 print(torch.cuda.current_device()) # 0 print(torch.cuda.is_available()) # TRUE 2.检查是否显存不足，尝试修改训练的batch size，修
动手学深度学习-pytorch-源代码
04-14
《动手学深度学习》是一本深受读者喜爱的书籍，它以PyTorch为工具，深入浅出地介绍了深度学习的基本概念、模型和实践方法。PyTorch是Facebook开源的一个强大的深度学习框架，以其易用性、灵活性和强大的计算能力而...
动手学深度学习-Pytorch版，练习jupter-nootbook源码
11-16
动手学深度学习-Pytorch版，练习jupter-nootbook源码。动手学深度学习-Pytorch版，练习jupter-nootbook源码。动手学深度学习-Pytorch版，练习jupter-nootbook源码。动手学深度学习-Pytorch版，练习jupter-nootbook...
Pytorch-pytorch深度学习教程之图像理解.zip
05-22
在本教程中，我们将深入探讨PyTorch在深度学习领域中的应用，特别是在图像理解方面的实践。PyTorch是一个强大的开源机器学习库，由Facebook的AI研究团队开发，它以其灵活性和易用性受到广大开发者和研究人员的喜爱。...
Yolov8模型训练报错：torch.cuda.OutOfMemoryError
大雾的小屋的博客
11-10 4818
最近在使用自己的数据训练Yolov8模型的时候遇到了很多错误，下面将逐一解答。 如何解决：torch.cuda.OutOfMemoryError的报错问题。
torch.cuda.OutOfMemoryError: CUDA out of memory.解决方案（亲测有效）
estxiyang的博客
04-12 2744
torch.cuda.OutOfMemoryError: CUDA out of memory.
完美解決pytorch载入预训练权重时出现的CUDA error: out of memory
狂小虎/Anthony Dave的博客 KuangXiaoHu's Blogs
09-11 3409
今天小虎远程操纵工作站，想把昨晚练好的预训练模型迁移一下，发现跟往常不一样，nvidia的Cuda报错说出现超出内存的情况。
CUDA out of memory（然鹅明明还有空间）
联邦学习小白
09-16 1万+
模型跑完想再跑一遍的时候居然给我报“CUDA out of memory”，很明显，我的显卡虽然菜，但起码还是可以跑的，毕竟刚跑完，那就可能是python在运行完以后没有释放资源，导致GPU的内存满了。 如果是这样的话，那解决的方法就很简单了，直接把进程杀死就行。 首先在C:\Program Files\NVIDIA Corporation\NVSMI位置打开命令行，然后输入nvidia-smi（也可以用nvidia-smi.exe -l 1，每秒更新CUDA使用情况），显示当前gpu的使用情况如下： 显
解决：RuntimeError: CUDA out of memory.
ln01010的博客
07-22 2230
利用pytorchGPU训练深度模型时出现RuntimeError:CUDA out of memory问题解决方案
CUDA out of memory的一种神奇解决方法
jasminefeng的博客
02-24 4138
我的输入样本维度是(1000,12,24,72)，一开始我是这么输入数据的： trainer.train(x_train, t_train, x_val, t_val) 发现必溢出，后来我取出其中400个样本输入： trainer.train(x_train[:400], t_train[:400], x_val, t_val) 发现不溢出了，训练正常，然后我把400删掉，但没删冒号： trainer.train(x_train[:], t_train[:], x_val, t_val) 竟然也没有溢
OutOfMemoryError（内存溢出）
u014754272的博客
03-15 2633
产生的原因： 1.内存中加载的数据量过于庞大，如一次从数据库取出过多数据。 2.集合类中有对对象的引用，使用完后未清空，使得JVM不能回收。,3, 3.代码中存在死循环或循环产生过多重复的对象实体。 4.使用的第三方软件中的BUG。 5.启动参数内存值设定的过小。 重点排查以下几点： 1.检查代码中是否有死循环或递归调用。 2.检查是否有大循环重复产生新对象实体。 3...
CUDA ~ RuntimeError: CUDA Out of memory解决
whaosoft143ai的博客
05-30 1749
一些可以尝试的解决“RuntimeError: CUDA Out of memory”的方案。在这些选项中，如果你使用的是预训练模型，则最容易和最有可能解决问题的选项是第一个。 whaosoft aiot http://143ai.com 因此，如果你在不将梯度归零的情况下进行 2 个半批次的训练，它们的梯度回累积，最终得到以目标有效批次大小相同的梯度。在训练循环中，我们需要使用计数器根据小batchsize大小进行更新，一旦它达到预设的目标，那就是我们将梯度归零时。在那之前，他们只是通过loss.back
torch.cuda.OutOfMemoryError: CUDA out of memory.的解决办法
12-02
当出现torch.cuda.OutOfMemoryError: CUDA out of memory错误时，可以尝试以下解决方法： 1.将batch_size改小，减少每次传入模型的数据量。 2.使用torch.no_grad()包装测试代码，以减少内存占用。 3.使用torch....

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值