Batch-Size下对CPU和GPU的一些影响

下面是跑YOLOv5进行训练时，将Batch-Size设置为不同值时的CPU和GPU利用率的情况。

可以看到，GPU显存利用率越小的时候，几乎就是CPU疯狂在工作的时候，而GPU上去了，CPU就会轻松很多哈。

 

当Batch-Size = 8时

 

当Batch-Size = 16时

 

当Batch-Size = 32时，占用了10G显存。

 

当Batch-Size = 64时，基本打满了，14.8G，接近16G，不能在增加了。