关于yolov5训练时报错cuda内存不足问题_训练ai遇到cuda内存不足

作者：羊村懒王 | 2024-03-07 04:05:56

踩

训练ai遇到cuda内存不足

RuntimeError: CUDA out of memory. Tried to allocate 4.08 GiB (GPU 0; 10.76 GiB total capacity; 414.97 MiB already allocated; 4.08 GiB free; 4.85 GiB reserved in total by PyTorch)

当时报这种错误，当然调小batch-size可以解决，但是当不想调小的时候可以把工作线程改小，就能正常训练了

在train中有如上图所示的超参数，之前我的工作线程是8，在同等条件下改为4就正常能训练了

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/203435?site