火辣潮女 发表于 2025-5-10 16:28:04

求助，yolov8跑dota数据集爆显存

rt，autodl的3080ti12g显存，把batchsize降到4才能开始跑，然后跑一轮在验证的时候爆，其他的batch直接是跑不了一点

花开の又是一季 发表于 2025-5-10 16:28:31

为什么直接复制后面的给吞了

断舍离 发表于 2025-5-10 16:28:46

接一楼，报错： CUDA out of memory. Tried to allocate 2.97 GiB (GPU 0; 11.76 GiB total capacity; 6.23 GiB already allocated; 2.45 GiB free; 7.37 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

#NAME? 发表于 2025-5-10 16:28:57





不入爱河 发表于 2025-5-10 16:29:03

我也遇到这个问题了，哥，怎么解决的。我用的dota1.0。每次验证阶段就会爆显存，我都设置work1了

知酒客 发表于 2025-5-10 16:29:10

好像得把图像切割

不会飞的娃纸 发表于 2025-5-14 20:34:57

啥也不说了，没空看，我正在发帖呢，没空，
页: [1]
查看完整版本: 求助，yolov8跑dota数据集爆显存