测试场景
我的电脑没有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
标签：20,Python311,16,GPC,yaml,内存,测试,视觉 From： https://www.cnblogs.com/harrychinese/p/18004880