机器视觉-无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

posted @ 2024-02-03 16:07  harrychinese  阅读(14)  评论(0编辑  收藏  举报