机器视觉-无GPC情况下增加内存的速度测试
测试场景
我的电脑没有GPU, 想着增加内存多少会提升一下深度学习的速度, 实践证明, 增加内存并不能提升速度, 连一星点效果都没有, 原因也简单, 瓶颈在CPU上而不是内存.
如果手上没有GPU的电脑, 还是直接在算力平台上租用靠谱.
测试结果如下:
内存 | batch | 内存使用率 | 5 epoch耗时 |
---|---|---|---|
16 | 16 | 80% | 20分 |
40 | 16 | 40% | 20分 |
40 | 32 | 80% | 20分 |
测试脚本
C:\Python311\Scripts\yolo.exe task=detect mode=train model=yolov8m.yaml data=C:\Python311\Scripts\datasets\pcb\data.yaml epochs=5 workers=1 imgsz=640 batch=32
数据集下载 https://universe.roboflow.com/1-tvs9l/-n3tdy/dataset/13