算
大模型训练集群的组网,InfiniBand 和 RoCE 是绕不开的二选一,聊聊决策逻辑。
InfiniBand 的优势
- RDMA 原生,延迟最低,集合通信(NCCL)优化最成熟;
- 无损网络有完善的拥塞控制;
- 大厂万卡集群的默认选项,出了问题好找人问。
RoCE 的优势
- 跑在标准以太网上,运维团队现成的以太网经验能用上;
- 交换机选择多、价格相对友好;
- 和现有数据中心网络融合容易,不用维护两套 fabric。
决策看这三点
1. 规模:千卡以上、AllReduce 频繁,IB 的确定性优势明显;几百卡、通信占比不高,RoCE v2 完全够用。
2. 团队:有 IB 运维经验就上 IB,没有的话 RoCE 的学习曲线平缓得多。
3. 预算:IB 网卡+交换机整体贵一截,RoCE 能省不少。
几个坑
- RoCE 一定要配 PFC + ECN,不然丢包重传能把训练拖死;
- 不管选哪个,拓扑先定好(胖树还是 Dragonfly),后期改拓扑等于重建;
- 网卡固件版本全集群统一,混用版本出过不少灵异问题。
你们集群用的哪种方案?踩过什么坑?
InfiniBand 的优势
- RDMA 原生,延迟最低,集合通信(NCCL)优化最成熟;
- 无损网络有完善的拥塞控制;
- 大厂万卡集群的默认选项,出了问题好找人问。
RoCE 的优势
- 跑在标准以太网上,运维团队现成的以太网经验能用上;
- 交换机选择多、价格相对友好;
- 和现有数据中心网络融合容易,不用维护两套 fabric。
决策看这三点
1. 规模:千卡以上、AllReduce 频繁,IB 的确定性优势明显;几百卡、通信占比不高,RoCE v2 完全够用。
2. 团队:有 IB 运维经验就上 IB,没有的话 RoCE 的学习曲线平缓得多。
3. 预算:IB 网卡+交换机整体贵一截,RoCE 能省不少。
几个坑
- RoCE 一定要配 PFC + ECN,不然丢包重传能把训练拖死;
- 不管选哪个,拓扑先定好(胖树还是 Dragonfly),后期改拓扑等于重建;
- 网卡固件版本全集群统一,混用版本出过不少灵异问题。
你们集群用的哪种方案?踩过什么坑?
