Skip to content

AI 集群整体架构设计:从需求到分层蓝图 ​

说明:本篇为综述/无第一手实验数据,采用集群架构设计公开实践。

实验目标 ​

集群架构是 AI-HPC 的"总纲":容量规划(02 章)、网络设计(03 章)、存储(04 章)等各章是零件,本篇把它们装成整机。目标:建立从需求到蓝图的完整流程——需求推导(规模/负载/可用性)→ 四平面分层(计算/存储/网络/管理)→ 平面间带宽配比 → 架构文档与设计检查清单。读完能回答:一个集群怎么从需求出发设计、各平面怎么分层、带宽怎么配比、架构文档写什么。

代码逐段拆解 ​

第一段:需求推导——从业务目标算集群规模 ​

python
def size_cluster(workloads, target, concurrency):
    # 以训练负载为主推导(估算口径):
    # 1) 单卡算力 × 卡数 = 总算力;以"目标模型规模/训练周期"反推卡数
    #    e.g. 7B 模型 × 1T tokens,需要 X 卡 × Y 天(compute-cost 公式,10 章)
    # 2) 并发项目数 × 单项目卡数 = 并行需求(多团队多任务)
    # 3) 推理/科学计算负载占比 → 附加需求(异构分区,02 章容量规划)
    # 4) 冗余:N+1 电源/网络,故障域设计
    total_gpus = model_compute(target) / (gpu_flops * days) + concurrency_gpus
    return plan(total_gpus)

要点:

  1. 规模从"目标模型 × 训练周期"反推:不是"买多少算力"而是"给定模型和周期需要多少卡"——10 章 compute-cost 公式是推导基础(估算口径)。
  2. 并行需求单独算:多团队多项目并发 = 项目数 × 单项目卡数,与训练需求叠加——容量规划(02 章)直接喂给架构设计。
  3. 冗余是设计约束不是可选项:电源/网络 N+1、故障域划分——规模推导后必须叠加冗余。

第二段:四平面分层与带宽配比 ​

python
# 四平面(各章衔接):
#   计算平面:GPU 节点 + 高速互联(03 章 IB/RoCE)
#   存储平面:共享存储(04 章并行文件系统)+ 数据加载网络
#   网络平面:计算网 / 存储网 / 管理网分离(03 章平面解耦)
#   管理平面:带外管理(BMC/IPMI)+ 监控(02 章 DCGM)
# 带宽配比(经验,估算口径):
#   计算网:1:1 无收敛(GPU 通信),万卡级 400G 互联
#   存储网:按"checkpoint 写盘窗口"设计(04 章:写满 GPU 显存 ≤ 分钟级)
#   管理网:低带宽(带外 + 监控遥测)

要点:

  • 四平面是架构骨架:计算/存储/网络/管理各司其职,各章知识在这里对齐——架构设计是"把各章零件装成整机"。
  • 带宽配比从"关键路径"反推:计算网按 AllReduce 无收敛配、存储网按 checkpoint 窗口配(04 章)、管理网低带宽——每根带宽都要有业务理由。
  • 平面分离是万卡前提:不分离则流量互相干扰(checkpoint 拖垮训练通信)——03 章平面解耦在此落地。

原理讲解 ​

架构分层与各章衔接图 ​

需求(模型/周期/并发/冗余)
  └── 计算平面:GPU 节点(01 硬件) + 互联(03 网络)
  └── 存储平面:并行文件系统(04 存储) + 数据管线(13 数据篇)
  └── 网络平面:三网分离(03) + 遥测(02 DCGM/网络监控)
  └── 管理平面:装机(05) + 调度(06 云原生) + 治理(02)

核心认知:集群架构不是"买一堆设备",而是需求驱动的分层设计——每一层的能力由下层支撑、为上层服务,各章是层内细节,架构是把层间关系定清楚。

关键设计决策清单 ​

决策问题关联章
规模卡数/算力如何推导02 容量规划、10 算力成本
互联IB or RoCE、收敛比03 选型
存储容量/带宽/checkpoint 窗口04 存储
冗余N+1、故障域、HA05 PXE HA、03 网络 HA
调度队列/分区/QoS06 云原生、07 异构

架构文档 = 决策 + 理由:每个决策写"选了什么 + 为什么 + 数字依据",架构文档才可评审、可演进。

实验结果 ​

(估算口径,非实测)

规模计算平面存储带宽网络收敛比
千卡400G 互联数百 GB/s计算网 1:1
万卡400G 分层(SU 单元)TB 级计算网 1:1 无收敛

工程启示 ​

  1. 需求先行:规模从"目标模型 × 周期"反推,容量规划(02 章)喂给架构;冗余是设计约束。
  2. 四平面分层:计算/存储/网络/管理各司其职,各章知识在架构层对齐——架构是"装整机"不是"买零件"。
  3. 带宽从关键路径反推:计算网按 AllReduce、存储网按 checkpoint 窗口、管理网低带宽——每根带宽有业务理由。
  4. 文档记录决策与理由:架构文档 = 每个决策"选什么 + 为什么 + 数字依据",可评审、可演进。

下一篇预告:架构总纲定稿,回到框架层——"模型并行详解:张量并行、流水线并行与序列并行"。

AI-HPC 智算超算专家 · Contact: openaihpc@gmail.com