分布式加速模板

要用 horovod 做多机多卡分布式训练加速时

05-任务模板 / 模板使用
分布式加速horovod分布式训练work_imagesworking_dircommandnum_workerring-allreduce

分布式加速

horovod

horovod 分布式训练模板。

入参:

  • work_images:worker 的运行镜像,如 horovod/horovod:0.20.0-tf2.3.0-torch1.6.0-mxnet1.5.0-py3.7-cpu
  • working_dir:命令的启动目录,如 /mnt/admin/pipeline/example/horovod/
  • command:训练启动命令,如 python /examples/tensorflow2_mnist.py
  • num_worker:分布式 worker 的数目,如 2
最后更新 2026-06-30完整文档以官方仓库为准:GitHub Wiki