下面是跑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，不能在增加了。
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。