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，不能在增加了。