最近在使用YOLOv3进行测试时候，会报错cuda out of memory，然后自己也尝试了一些方法，比如缩小测试图片尺寸等等，但是问题依然没有解决。想问一下各位大佬，有没有什么好的解决办法，或者YOLOv3有没有提供一些函数来释放显存，就是模型用完以后直接释放掉。谢谢谢！！！
最近在使用YOLOv3进行测试时候，会报错cuda out of memory
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
分享- 邀请回答
-
2条回答 默认 最新
- 2022-04-12 15:51回答 2 已采纳 1 batchsize调到8 也是一样的报错。2 训练前我用nvidia-smi查看GPU显存，已使用0.121G（总共6G），一旦开始训练已使用显存数据就开始上升，最终到5.359G后程
- 2022-04-12 16:34回答 1 已采纳 降低batchsize的大小，例如8 16
- 2022-09-04 14:25回答 1 已采纳 从6月初开始，6G显存的显卡开始出现CUDA Error:out of memory的问题，这是因为dag文件一直在增加，提示显存不足，不过要增加到6G还需要最少两年的时间。现在出现问题的原因是1、内
- 2023-05-13 14:13_我的旺仔呢_的博客 内存炸了，重启或者减小workers、batch-size参数。
- 2023-04-04 09:44回答 4 已采纳 该回答通过自己思路及引用到各个渠道搜索综合及思考,得到内容具体如下： 根据您提供的信息，这个错误是由于GPU内存不足引起的。在训练yolov7标准版时，需要更多的GPU内存，而您的GPU内存已经被占用
- 2022-03-06 00:23回答 2 已采纳 你这个是显存满了
- 2022-07-21 14:09回答 2 已采纳 在这个网址中，查看DeepStream-Yolo需要那些配置https://github.com/marcoslucianops/DeepStream-Yolo#yolov5-usage在Requir
- 2023-08-15 22:27浪浪山猪的博客 换个同样配置的电脑能跑，昨天...最后发现是环境问题，应该是由于昨天在开发的时候不知道加了什么模块进去导致的冲突，我一直又备份环境的习惯，把最初的环境重新放到anaconda的env中，删掉了有问题的环境，可以跑了。
- 2022-05-21 14:16回答 1 已采纳 cuda0,指的是你gpu的编号，就是说使用的是你的第0张显卡。一般用一张显卡都是这样的。
- 2023-01-31 00:52
使用DWY100k数据集对UEA进行测试，出现报错：IndexError: index 125000 is out of bounds for axis 0 with size 95500
python
tensorflow
深度学习
回答 4 已采纳 这个错误是由于使用的数据集大小不匹配导致的。报错提示的 index 12500 超出了索引范围，数组大小为 95500。 解决方案是检查数据集的大小是否与代码预期的大小匹配，并修正代码中的数据集引用， - 2023-02-10 00:52回答 6 已采纳 你这显示你的DarkNet中的cudnn是8.1.1的版本，这个和你说的8.6对不上。不过按理来说cudnn的影响不会很大（只要不是cudnn7.6.5和cudnn8.x这种大版本大区别就基本上没啥大
- 2022-11-04 00:39Yong Hui的博客 RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.66 GiB already allocated; 0 bytes free; 1.72 GiB reserved in total by PyTorch) If reserved memory is >>...
- 2022-09-09 13:24回答 2 已采纳 我记得之前我看issue，美团官方是提了一个PR，解决了这个bug，如果你这里的代码是最新版本的还是有问题，请联系美团官方。 Resume Training issue:
- 2022-03-14 23:47AI视觉网奇的博客 yolov5 CUDA out of memory torch版本1.8，后来发现原因是： workers大于1了，可能开了多进程的缘故。 parser.add_argument('--workers', type=int, default=1, help='maximum number of dataloader workers') ...
- 2022-11-25 15:10WWW_lena的博客 改进yolov7时出现显存不够的问题
- 没有解决我的问题, 去提问
