NNetpath
原理深入预计 30 分钟

MPI 与集合通信:谁在真正搬数据

MPI 在 AI 训练里常常只是个启动器,真正的通信由 NCCL 完成。搞清分工才能调对。

学完这节你能做到

  • 说清 MPI 与 NCCL 在 AI 训练里的分工
  • 看懂 mpirun 常用参数对性能的影响
  • 排查 hcoll 与 NCCL 冲突这类初始化失败

建议先学跳过这几节会看不懂本节的部分推导

正文待编写以下是本节已定稿的小节大纲
  1. 1MPI 基础:rank、communicator、集合操作
  2. 2OpenMPI 与 UCX 的分层
  3. 3mpirun 关键参数:-np、-bind-to、-x
  4. 4MPI 只做启动器的场景
  5. 5hcoll 与 NCCL 抢同一条路的典型报错
  6. 6与 Kubeflow Trainer / MPIJob 的结合

延伸资料

  • ·k8s-in-actionai/nccl-tests/config-reference.md