AI 网络概述 · 人工智能 · 技术 · 文档 · CodingBox
CodingBox 文档

AI 网络概述

训练大型 AI 模型把计算负载分摊到许多服务器上的许多块 GPU 上。它们之间的网络——而不是任何一块单独的 GPU——往往决定了集群的训练速度,这也是 AI 组网倾向于使用能获得的带宽最高的光模块的原因。

为什么网络至关重要

  • 集合操作——GPU 之间持续以 all-to-all 模式交换梯度和参数,产生大量 东西向流量。
  • 尾延迟至关重要——每一步都要等待最慢的链路,因此稳定的低延迟和无损传输是必须的。
  • 规模——数千块 GPU 意味着高密度、高基数的交换机以及庞大的光模块数量。

两种组网方式

  • InfiniBand——在 HPC 领域历史悠久,具备 RDMA 和无损流控(见 InfiniBand)。
  • 带 RoCE 的以太网——即融合以太网上的 RDMA,在大规模 AI 场景中的应用日益增多,运行在无损以太网上。

光模块

AI 链路通常是 400G 和 800G,采用 QSFP-DD 和 OSFP 模块。具体的互连方案和光模块选型见 AI 互连与光模块。

同样的封装形式和管理存储器在这里同样适用——CodingBox 读取这些高速模块的方式与其他模块并无不同。

延伸阅读

  • AI 集群布线——每块 GPU 的链路数、 rail-optimised 布局中的距离、DAC/AEC/AOC/DR 组合、连接器选型、光模块功耗、安装 QA。
  • AI 组网中的链路可靠性与监测—— 为什么一条链路会拖垮整个任务、FIT 算法、预测性指标、监测架构、隔离与预防。

如果您发现本资料中有不准确之处或错误,请选中相应片段并按 Ctrl+Enter,即可。