Multi-GPU Training多GPU训练（Yolo系列）

单 GPU

python train.py  --batch 64 --data coco.yaml --weights yolov5s.pt --device 0

多 GPU 数据并行模式

python train.py  --batch 64 --data coco.yaml --weights yolov5s.pt --device 0,1

多GPU DistributedData并行模式（✅推荐）

python -m torch.distributed.run --nproc_per_node 2 train.py --batch 64 --data coco.yaml --weights yolov5s.pt --device 0,1

使用特定 GPU

通过简单地传递后跟您的特定 GPU 来做到这一点。例如，在下面的代码中，我们将使用 GPU .--device2,3

python -m torch.distributed.run --nproc_per_node 2 train.py --batch 64 --data coco.yaml --cfg yolov5s.yaml --weights '' --device 2,3