yolov5单机多卡训练加速配置

用默认单卡训练速度太慢,这里配置多卡,并使用缓存,内存得够用要不然无法缓存

--device 0,1 --cache

yolov5单机多卡训练加速配置_第1张图片

--device 0,1这里的0就是显卡1,1就是显卡2,如果你有更多显卡,可以继续往下写

训练过程

yolov5单机多卡训练加速配置_第2张图片

通过GPU-Z查看2个显卡的占用率

显卡1

yolov5单机多卡训练加速配置_第3张图片

显卡2

yolov5单机多卡训练加速配置_第4张图片

你可能感兴趣的:(YOLO)