Skip to content

AI 集群网络总览 ​

一句话理解:AI 集群网络是为"成千上万颗加速器在同步屏障内同时说话"而设计的专用 fabric——scale-up 域(NVLink/UALink)负责机柜内的内存语义互联,scale-out 域(InfiniBand/RoCE/UEC 以太网)负责机柜间的大象流与 all-to-all,两者的带宽、时延、无损语义与故障域设计完全不同,不能用传统云网络的思想硬套。

为什么 AI 平台工程师必须懂 AI 集群网络 ​

不懂 AI 集群网络的工程师,很容易把集群建成"单卡 benchmark 很好看、一上分布式就翻车"的状态。典型场景:

  • 128 卡训练的 NCCL all-reduce 实测带宽只有标称的 60%,换块光纤就好了——这是 光模块/FEC 误码导致的链路降速,监控里早就报了计数。
  • 千卡训练每隔几小时随机挂一个 rank,NCCL timeout——这是 PFC 风暴 + ECN 参数不当造成的无损域失稳,不是 GPU 坏了。
  • MoE 模型一打开专家并行,网络吞吐打满、step 时间翻倍——这是 all-to-all 流量的 incast没有被自适应路由和拥塞控制兜住。
  • 采购了 800G 交换机,实测却只有 400G 的有效吞吐——这是 ECMP 哈希极化:大象流数量太少,哈希把多条流钉在了同一条上联上。
  • 集群从 4k 卡扩到 24k 卡,不改动任何配置,性能从 90%+ 掉到 10–90% 不等——这是 Meta 2024 年真实踩过的坑:规模本身会暴露拓扑感知调度、路由、集合通信算法三层问题。

这些都不是"买更快的交换机"能解决的。AI workload 对网络的需求(同步屏障、尾部延迟、大象流、无损语义)与传统云网络(多租户、南北向、尽力而为)几乎是反着的——理解 scale-up/scale-out 的二分、带宽时延比、incast 与拥塞控制、拓扑与故障域,是设计生产级 AI 集群的必修课。

与「计算机网络」主题的分工 ​

本手册 计算机网络 主题已经讲了通用网络原理:OSI/TCP 分层、TCP 拥塞控制、RDMA/RoCE/InfiniBand 的基本概念、数据中心拓扑(Spine-Leaf/Clos/fat-tree/OCS)的通识。本主题不重复这些内容,聚焦 AI 集群专用的网络工程:

维度计算机网络(通用原理)本主题(AI 集群专用工程)
视角协议与分层:TCP/RDMA 怎么工作系统与取舍:训练/推理 workload 如何重塑 fabric 设计
范围通用数据中心 + 云网络GPU 集群的 scale-up 域与 scale-out 域
重点可靠性、通用性、多租户尾部延迟、incast、无损语义、网内计算、故障域
案例通用云场景Meta/xAI/阿里/腾讯/Google/OpenAI 的量化实践

学习目标 ​

阅读完本主题后,你应该能够:

  1. 解释 scale-up 与 scale-out 的划分依据,说出 NVLink 5(1.8 TB/s)、InfiniBand NDR/XDR(400G/800G)、RoCE 在带宽、时延、语义上的数量级差异。
  2. 从并行策略(TP/DP/EP/PP)推导一次训练 step 的流量矩阵:谁是大象流、谁是 incast、谁必须留在 scale-up 域。
  3. 画出典型 AI 集群网络全景:NVLink 域 → 节点 NIC → leaf-spine fabric → super spine,以及前端网、存储网、带外管理网的位置。
  4. 比较 InfiniBand、RoCEv2(PFC/ECN/DCQCN)、Spectrum-X、UEC 的技术路线:无损怎么实现、拥塞控制谁来做、网内计算有没有。
  5. 读懂 ring/tree all-reduce 与 all-to-all 的步数和字节数模型,用 α-β 模型解释"小消息 tree 胜、大消息 ring 胜"。
  6. 在生产中排查典型故障:all-reduce 掉带宽、NCCL timeout、PFC 死锁、哈希极化、光模块误码。
  7. 跟踪 2025–2026 的格局变化:UEC 1.x 规范、UALink 1.0、NVLink Fusion、Spectrum-X Photonics/CPO、OpenAI MRC。
  8. 回答初 / 中 / 高级 AI 集群网络面试题,把流量模型、拓扑与故障排查联系起来。

本主题与其他主题的关系 ​

主题解决的核心问题与本主题的关系
计算机网络通用网络原理:TCP/RDMA/拓扑/OSI本主题的底层原理;RDMA/RoCE/IB/fat-tree 的通用概念在那里
分布式系统基础并行计算模型、同步与一致性数据并行/流水并行的语义来源;本主题讲这些语义如何落到网络上
GPU 架构与 CUDAGPU 硬件、CUDA 编程模型NVLink/NVSwitch 的硬件载体;GPUDirect RDMA 的起点
CNI / CSI 深度K8s 网络与存储插件集群内 Pod 网络、多网卡接入、RDMA 设备插件化
GPU 调度与资源管理GPU 如何被调度与治理拓扑感知调度决定训练 Job 落在哪些 NVLink 域/轨(rail)上
vLLM / SGLang单实例推理引擎PD 分离、多机 TP 推理的通信模式复用本主题的流量模型
AI SRE可观测性与 SLO网络 SLI(带宽达标率、PFC pause、FEC 误码)是 AI SRE 的核心指标
xAI 案例Spectrum-X 的规模化验证95% 实测吞吐 vs 传统以太网 60% 的对照口径来源
DeepSeek 案例训练效率极限优化多平面网络拓扑、节点受限路由的设计动机
Google 案例TPU 集群与 OCSFalcon 传输、Jupiter OCS、ICI 拓扑与故障恢复
OpenAI 案例Stargate 多园区架构MRC 开放协议与超大规模 scale-out fabric

本章结构 ​

  1. 背景 — AI workload 的流量特征与传统云网络的本质区别:同步屏障、大象流、incast、尾部延迟。
  2. 核心思想 — scale-up/scale-out 二分、带宽时延比、无阻塞与收敛、rail-optimized、有损/无损/第三种路线。
  3. 架构设计 — 典型 AI 集群网络全景:从 GPU 到 super spine 的四层 fabric 与拓扑选型决策树。
  4. 工作流程 — 一次大规模训练的网络流量生命周期:从 NCCL 初始化到 step 内通信时序再到 checkpoint。
  5. 核心模块 — NVSwitch、IB 子网管理器、拥塞控制(PFC/ECN/DCQCN/自适应路由)、SHARP 网内计算、遥测与故障域。
  6. 生态与源码分析 — InfiniBand vs RoCE vs UEC 对比表;NCCL 源码结构、UEC 规范与 Google Falcon 设计剖析。
  7. 工程实践 — 纯 Python 可运行的 AI 集群网络模拟器(拓扑、流量矩阵、incast/DCQCN、PFC 死锁、集合通信模型)。
  8. 企业生产实践 — 选型决策、Meta/阿里/腾讯/xAI 量化实践、排障 playbook、扩展路径。
  9. 最佳实践 — fabric 设计检查清单、PFC/ECN 参数模板、NCCL 环境变量与验收标准。
  10. 面试题 — 初 / 中 / 高级 AI 集群网络面试题。
  11. 延伸阅读 — 官方文档、论文、一手发布与手册内回链。

一句话总结 ​

AI 集群网络的精髓在于**"把同步的集合通信当作一等公民来设计 fabric"**:scale-up 域用内存语义的专用互联(NVLink/UALink)换取微秒级时延,scale-out 域在无阻塞拓扑(fat-tree/rail-optimized)上以太网化(RoCE/Spectrum-X/UEC)换取开放与规模;对 AI 平台而言,选对协议路线决定生态与成本,调对拥塞控制决定利用率,管好光纤与故障域决定有效训练时间——三者共同决定一张昂贵的 GPU 集群能不能把理论算力变成模型训练进度。

参考来源

Released under CC-BY-SA-4.0 License.