1. InfiniBand技术全景与NVIDIA产品定位

在AI大模型训练与HPC仿真场景中,计算节点的算力往往不是瓶颈,真正决定集群效率的是节点之间的互联网络。当单卡算力进入PetaFLOPS量级、单集群规模扩展到数千甚至上万张GPU时,网络需要同时满足三个苛刻条件:微秒级延迟、接近线速的带宽利用率、以及极低的CPU占用。InfiniBand(IB)正是为满足这类需求而设计的网络技术体系,而NVIDIA通过收购Mellanox,把IB从传统HPC互联推向了AI集群的事实标准。

从HPC到AI:InfiniBand的演进逻辑

InfiniBand最早由IBTA在2000年前后标准化,目标是替代PCI总线,成为服务器内部与集群级的统一互联。它采用基于信用的流控、通道化交换和硬件卸载三大机制,从协议层就为低延迟和高吞吐服务。早期IB主要服务于气象、流体、分子动力学等MPI密集型HPC应用,典型特征是“节点多、消息小、通信频繁”。

进入AI时代后,负载特征发生了明显变化:AllReduce、AllGather等集合通信成为主流,单次通信数据量大、参与GPU多、对带宽和尾延迟极其敏感。以GPT类模型训练为例,一次梯度同步可能涉及数百MB到数GB的数据交换,若网络带宽不足或拥塞控制不佳,GPU利用率会从90%以上迅速跌到50%以下。IB的RDMA(Remote Direct Memory Access)能力恰好匹配这一需求:数据可以直接从一台机器的用户态内存搬到另一台机器的用户态内存,绕过内核协议栈,不占用CPU,也不需要多次内存拷贝。

RDMA为什么是核心优势

传统TCP/IP通信路径是:应用缓冲区 → 内核Socket缓冲区 → 网卡 → 网络 → 对端网卡 → 内核缓冲区 → 应用缓冲区,中间涉及多次上下文切换和内存拷贝。RDMA把这套路径压缩为:应用注册内存 → 网卡直接读取 → 网络 → 对端网卡直接写入目标内存。其直接收益包括:

  • 低延迟:典型IB网络端到端延迟可低至1~2微秒,而TCP/IP通常在几十微秒量级。
  • 低CPU占用:通信过程由HCA(Host Channel Adapter)硬件完成,CPU可专注于计算。
  • 高带宽利用率:配合RoCE或原生IB,可接近线速运行,减少协议开销。
  • 零拷贝:避免用户态与内核态之间的数据搬运,降低内存带宽压力。

需要区分的是,RDMA是一种能力,InfiniBand是承载RDMA的原生网络体系;RoCE则是在以太网上承载RDMA的方案。IB在拥塞控制、自适应路由、服务质量上更成熟,适合对确定性要求高的AI训练集群。

NVIDIA三条产品线:Quantum、ConnectX、BlueField

NVIDIA的IB产品线可以按“网络中的位置”来理解:ConnectX是网卡,装在服务器和GPU节点上;Quantum是交换机,负责把节点连成Fabric;BlueField是DPU,把网卡能力与可编程ARM核结合,承担存储、安全、虚拟化卸载。三者共同构成从端到网的完整链路。

以当前主流代际为例,ConnectX-7支持最高400Gb/s单端口速率,ConnectX-8进一步向800Gb/s演进;Quantum-2交换机提供32×400Gb/s或64×200Gb/s端口,采用液冷或风冷形态;Quantum-3则面向800Gb/s时代。BlueField-3 DPU集成16核ARM CPU和400Gb/s网络接口,可在网卡侧运行防火墙、存储虚拟化和集合通信卸载。

选型时不能只看端口速率,还要看以下维度:

  • GPU与网卡配比:常见做法是每张GPU配一块200Gb/s或400Gb/s网卡,训练集群倾向于1:1甚至更高。
  • 交换机端口密度与收敛比:AI集群通常要求无收敛(1:1),即下行端口总带宽等于上行端口总带宽。
  • 是否使用DPU:若需要存储卸载、多租户隔离或云化部署,BlueField更合适;纯计算互联则ConnectX足够。
  • 线缆与光模块:400Gb/s常用QSFP112或OSFP,短距用DAC,长距用AOC或光模块。
选型顺序建议:先确定GPU数量和通信模式,再定网卡速率与数量,最后反推交换机端口和拓扑。不要先买交换机再凑节点。

一张图看清端到端链路

下图把“GPU—网卡—交换机—对端”的RDMA数据路径与三条产品线的位置对应起来,便于建立整体定位。

InfiniBand 端到端 RDMA 链路与 NVIDIA 产品定位 计算节点 A GPU 显存 / 用户态缓冲区 已注册 MR(Memory Region) ConnectX 网卡(HCA) RDMA 引擎 / QP / CQ 200/400/800 Gb/s BlueField DPU(可选) 存储/安全/虚拟化卸载 Quantum 交换机 InfiniBand Fabric 核心 自适应路由 / 拥塞控制 SHARP 集合通信卸载 32×400G / 64×200G 端口 计算节点 B GPU 显存 / 用户态缓冲区 目标 MR 直接写入 ConnectX 网卡(HCA) RDMA 引擎 / QP / CQ 200/400/800 Gb/s BlueField DPU(可选) 存储/安全/虚拟化卸载 RDMA Write 直接写入 ACK / CQE 数据路径:GPU 内存 → ConnectX HCA → Quantum Fabric → 对端 ConnectX HCA → 对端 GPU 内存,全程绕过内核与 CPU

选型依据与常见误区

实际项目中,选型最容易犯的错误是“只看端口速率,不看收敛比”。例如一个128节点集群,每节点8张400Gb/s网卡,若采用1:1无收敛,核心层需要提供128×8=1024个400Gb/s下行端口,通常需要多台Quantum交换机组成两级Clos。若为了省钱做成3:1收敛,训练时AllReduce流量会在上行链路排队,尾延迟显著上升,GPU利用率下降。

另一个误区是忽略线缆和光模块的兼容性。400Gb/s IB常用QSFP112封装,与以太网的QSFP-DD并不完全通用;DAC长度一般不超过3米,超过后应改用AOC或有源光缆。采购前应确认交换机端口封装、网卡端口封装和线缆三者一致。

对于以存储访问为主的场景,BlueField DPU可以把NVMe over Fabrics、加密和虚拟交换从主机CPU卸载,释放出的核心可用于训练任务。但若集群规模较小、业务单一,ConnectX加普通交换机即可满足,不必为DPU支付额外成本。

一句话总结:ConnectX决定单节点通信能力,Quantum决定Fabric的规模和确定性,BlueField决定网络能否承担更多基础设施职能。三者按需组合,才是合理的IB集群选型。