yolov4-pytorch icon indicating copy to clipboard operation
yolov4-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

24GB显存不足按理啊来说不应该呀

Open jccmoing-bit opened this issue 2 years ago • 2 comments

训练的时候batch_size从64降到16都是显示显存不足的情况？？ CUDA out of memory. Tried to allocate 46.00 MiB (GPU 0; 24.00 GiB total capacity; 21.55 GiB already allocated; 32.38 MiB free; 21.74 GiB reserved in total by PyTorch 有大佬能告诉我这个咋回事嘛？

jccmoing-bit avatar Apr 04 '22 02:04 jccmoing-bit

再调小一点，我的也是24GBatch_size16也显存不够

Fan-tas-tic avatar Apr 06 '22 10:04 Fan-tas-tic

yolov4大得很……

bubbliiiing avatar Apr 08 '22 15:04 bubbliiiing