关于YOLO多机多卡训练

Question

想要了解一下，YOLO要怎么做到多机多卡训练？目前的硬件环境是学校机房（科研用），大概有30台，每一台的GPU配置是GTX1050，Win10，有两个服务器可用。

Answer 1

目标检测 YOLOv5 - 多机多卡训练_store = tcpstore(master_addr, master_port, world_s_文野历笑生的博客-CSDN博客 目标检测 YOLOv5 - 多机多卡训练flyfish环境：Python>=3.6.0 PyTorch>=1.7YOLOv5:v5分两种类型机器一个master，其他都是slavemaster机器运行的命令python -m torch.distributed.launch --nproc_per_node G --nnodes N --node_rank 0 --master_addr "192.168.1.2" --master_port 1234 train.py --b_store = tcpstore(master_addr, master_port, world_size, start_daemon, timeout https://blog.csdn.net/flyfish1986/article/details/119786227

【深度学习】【YoloV5】单机多卡训练 多机多卡训练_yolov5多卡训练-CSDN博客 官网给的： https://github.com/ultralytics/yolov5/issues/475指定单显卡：1 使用pytorch 的DataParallel mode 指定多显卡（不推荐），为啥不推荐，因为反传的时候梯度默认都在0卡上算了，所以显存分配不均衡，在显卡0占用非常多，别的显卡的显存又占用没那么多。训练时间也没增加多少。1 单机多卡(推荐) 使用pytorch的 DistributedDataParallel Mode 进行单机多卡训练。(推荐)1 –nproc_per_no_yolov5多卡训练 https://blog.csdn.net/weixin_43960370/article/details/129943281

《YOLOv5全面解析教程》六，YOLOv5使用教程详解（单卡，多卡，多机训练） - 知乎 前言 代码仓库地址：https://github.com/Oneflow-Inc/one-yolov5 欢迎star one-yolov5项目 获取 最新的动态。 如果你有问题，欢迎在仓库给我们提出宝贵的意见。 如果对你有帮助，欢迎来给我Star呀 ~ 声明:Mo… https://zhuanlan.zhihu.com/p/585174594