训练模型时GPU占用率过低

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

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

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

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

猜你喜欢

转载自blog.csdn.net/H_XKA_H/article/details/131391288