yolov8 多卡训练
时间: 2023-11-02 14:07:55 浏览: 421
2-20horovod多机多卡训练环境配置+训练流程1
yolov8的多卡训练非常简单,只需要在命令行中添加参数"device='0,1,2,3'"即可。具体的命令行指令如下所示:
yolo task=detect mode=train model=yolov8n.pt data=data/fall.yaml batch=32 epochs=100 imgsz=640 workers=16 device=\'0,1,2,3\'
其中,"device='0,1,2,3'"表示使用4张显卡进行训练。需要注意的是,这里的引号和反斜杠是必须的,否则命令行会出现语法错误。
阅读全文