YOLOv5训练速度慢 GPU占用率低_yolo训练gpu利用率低

作者：盐析白兔 | 2024-02-18 00:20:54

踩

yolo训练gpu利用率低

用Yolo5跑自己的数据集，torch_GPU和CUDA都安装了，使用GPU跑模型，GPU的占用率为1%，设置的epoch为300，batch size为32，大概6min多/epoch，数据集很小，刚过一千。在网上查了相关的文章，发现可能是读取数据造成的，把cache-image的action='store_true'改成'store_false'。

改成

修改后每epoch大概是13s！超级快，GPU的占用率变到33%，CPU的占用率50%。

后面再想办法改一改，计算机小白学习好难，论文好难（哀嚎~）！

声明：本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：【wpsshop博客】