训练模型时GPU占用率过低_yolov8训练时gpu利用率低

作者：2023面试高手 | 2024-06-13 02:49:34

踩

yolov8训练时gpu利用率低

训练yolov8模型时发现GPU利用率过低

训练模型时设置的batch-size是12，worker是3，按照网上的方法将batch-size或者worker调大，会出现内存不足和CUDA相关的各种问题，最后才发现训练模型时的GPU利用率，不是在任务管理器查看的。

cmd打开终端，在终端输入nvidia-smi，回车执行命令就能看到GPU利用率：

90%，怪不得调高batch会内存爆掉。

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