pytorch 在运行代码时，报错CUDA out of memory时候，的一些解决方法

最新推荐文章于 2024-04-24 16:26:12 发布
最新推荐文章于 2024-04-24 16:26:12 发布
阅读量1.1k 收藏 3
点赞数 1
分类专栏： 学习笔记 文章标签： 图像处理 计算机视觉 机器学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_48127442/article/details/115217604
版权

pytorch 在运行代码时，报错CUDA out of memory时候，的一些解决方法
运行代码报错为内存不足时

方法一：修改bachsize的大小。

bachsize越大，需要的空间越大，所以应该将bachsize适当调小一点。在这里插入图片描述

方法二：修改每次处理的图片数量。

batchimage：每次处理的图片数量越多，则需要分配越多的内存，因此，当在自己的笔记本上跑小的模型的时候，出现内存错误，可以将batchimage的数目调小一点，这样需要的内存则要少一些。在这里插入图片描述

方法三：修改数据集大小

如果在笔记本上跑代码时候，出现以上的内存不足问题时候，可以采用将数据集分出一个小的来进行修改。比如原数据集是10000张图70种类型的图，可以自己分出一个只含有两种图片的很小的数据集。
（以上解决方法都是借鉴师兄的。）

杜菜菜呀
关注
  • 1
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
成功解决torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 86.00 MiB (GPU 0； 2.00 GiB to
近期请国内外头部出版社可尽快私信博主！——心比天高，仗剑走天涯，保持热爱，奔赴向梦想！低调，谦虚，自律，反思，成长，还算是比较正能量的博主，公益免费传播……内心特别想在AI界做出一些可以推进历史进程影响力的东西(兴趣使然，有点小情怀，也有点使命感呀)…
06-02 1万+
成功解决torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 86.00 MiB (GPU 0; 2.00 GiB total capacity; 1.67 GiB already allocated; 0 bytes free; 1.67 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_spl
Pytorch GPU显存充足却显示out of memory的解决方式
09-18
今天小编就为大家分享一篇Pytorch GPU显存充足却显示out of memory的解决方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Pytorch遇到的问题及解决方案 - 更新ing
热门推荐
Tsingzao的博客
01-17 4万+
1、安装完成后，不能import torch，提示 ImportError: dlopen: cannot load any more object with static TLS 解决办法：有很多答案都说是将import torch放在import cv2之前，但我试了之后还是不能解决，最后是通过在jupyter notebook中可以直接import torch。我是通过mobarxterm...
CUDA out of memory 报错解决方案
最新发布
m0_47867419的博客
04-24 779
介绍了CUDA out of memory的一种解决办法
pytorch遇见RuntimeError: CUDA out of memory的解决
橙子Blue不加冰
07-26 7832
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.
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
weixin_39326879的博客
05-15 5963
出现的原因 这个提醒的意思就是运行过程中所需的内存超过CUDA的内存 解决办法 1.最直接的解决办法是减小batch_size(常用),或者减小模型参数和输入大小(一般很少这样做); 2.如果测试过程遇到这种情况,加上 with torch.no_grad(): 内存就不会分配参数梯度的空间 3.如果在训练过程遇到这种情况,可以尝试在训练前先释放CUDA内存 nvidia-smi查看GPU使用率,如果使用率不高,就使用torch.cuda.empty_cache()释放内存 官网对于torch.cuda.e
torch.cuda.OutOfMemoryError
power的专栏
05-31 7004
遇到这个错误，按网上改小batch_size改的很小了，依然报错。错误原因是网络的输入输出shape不一致。后改小了网络结构，搞定。
解决Pytorch 训练与测试时爆显存(out of memory)的问题
09-18
今天小编就为大家分享一篇解决Pytorch 训练与测试时爆显存(out of memory)的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
【兼容调试】pytorch出现RuntimeError: CUDA out of memory时的一些解决方法
changreal的博客
02-12 5395
代码方面 要适当删除没用的中间变量（虽然效果不大） 比如：del output train的时候，累加loss 一定要+=loss.item()，不然会把loss的history也加进去，又占显存了 同理2，及时detach() 比如rnn decoder的输出中把上一个时间步的输出作为下一个时间步的输入，那么这个输入完全就可以.detach() valid or test的时候，一...
torch.cuda.OutOfMemoryError: CUDA out of memory.解决方案（亲测有效）
estxiyang的博客
04-12 2156
torch.cuda.OutOfMemoryError: CUDA out of memory.
已解决：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解决 RuntimeError: CUDA out of memory.
weixin_44505185的博客
11-08 1334
Pytorch解决 RuntimeError: CUDA out of memory.
搞深度学习，跑程序遇到 out of memory的问题
qq_39418067的博客
03-03 4821
我的系统是Ubuntu16.04，显卡是rtx 2080ti,跑ssd程序，结果报错 out of memory，像网上说的更改了batch_size ，改的很小也没有解决，还是会超出，考虑是不是够被占用了太多，都没有有效的释放才这样的，然后按下面办法进行了解决。 解决办法是：1.输入命令查看自己gpu的使用情况：watch -n 10 nvidia-smi 该命令可以监视gpu的动态，其中的数字...
CUDA报错:Out of Memory
Coldlebron的博客
10-28 1万+
RuntimeError: CUDA out of memory. Tried to allocate xx GiB (GPU 0; xx GiB total capacity; xx GiB already allocated; xx GiB free; xx GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid
一文读懂 PyTorch 显存管理机制
zandaoguang的博客
04-08 3887
点击上方“视学算法”，选择加"星标"或“置顶”重磅干货，第一时间送达作者丨米阿罗@知乎（已授权）来源丨https://zhuanlan.zhihu.com/p/486360176编辑丨极市平台首发于踢翻炼丹炉：https://www.zhihu.com/column/c_1320691511223136256二次转载须经作者授权导读本文细致的对PyTorch 显存管理机...
解决出现CUDA error:out of memory的问题
qq_46542320的博客
05-24 1万+
使用：CUDA_VISIBLE_DEVICES限制一下使用的GPU。CUDA_VISIBLE_DEVICES=0,1 则使用这两张GPU。CUDA_VISIBLE_DEVICES=0 则使用第一张GPU。CUDA_VISIBLE_DEVICES=1，则使用第二张GPU。如何设置CUDA_VISIBLE_DEVICES？例如：有两块GPU，即0,1号GPU，查阅资料，发现是显卡内存不足。
pytorch cuda out of memory
06-28
PyTorch CUDA 内存不足 这个错误通常是由于您的 GPU 内存不足而导致的。您可以尝试以下方法来解决这个问题： 1. 减少批量大小：减少每个批次中的样本...希望这些方法能够帮助您解决 PyTorch CUDA 内存不足的问题。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值