第 18 章 AI InfraGPU分布式训练

第 18 章 AI Infra 实践

第18章 AI Infra 实践

本章从工程实践视角梳理千卡/万卡 GPU 集群的完整建设路径。技术栈沿五层架构组织,本节先给出集群架构与基础设施 的实战决策——规模规划、硬件选型与网络选型,后续小节依次覆盖平台调度、分布式训练、数据工程与存储、可观测 性、统一可视化与元数据追溯,最后汇总运维实战与业界标杆。本节只陈述决策要点与工程取舍。

18.1 集群架构与基础设施实战

本节围绕集群落地的三个核心决策展开:建多大规模、用什么硬件、用什么网络。每个决策先给结论,再给判断依据;原 理性推导见对应章节。调度器需感知 NVLink 拓扑、训练框架需按网络拓扑选并行策略、存储经 GDS 直连硬件、可观测 系统收集各层指标,这些层间协作细节在后续小节逐一展开。

18.1.1 规模规划

模型参数从数亿增长到数千亿后,单卡显存与算力无法承载完整模型。Llama 3 405B 仅 BF16 权重就需约 810 GB 显存, 远超单卡 80 GB 容量,实际训练集群普遍达到千卡乃至万卡级别。 工业界按规模将集群分级,如表18-1所示。 表18-1 集群规模分级 集群规 GPU 数量 典型用途 代表案例 模 百卡级 64256 小型训练、微调、推理 企业私有化部署 千卡级 5124,096 中等规模预训练(约70B 模型) Llama 3 70B(约2,000 GPU) 万卡级 8,19232,76 大规模预训练(100B1T 模型) Llama 3 405B(16,384 GPU)、NVIDIA Eos(4,608 GPU) 十万卡 >50,000 超大规模训练、多模型并行 xAI Colossus(200,000 GPU)、Meta(350,000+ H100 级 equivalent) 本书讨论的千卡/万卡集群指 1,024 至 100,000+ 张 GPU 的训练基础设施。集群规模决定模型能力天花板、训练速度与实 验并行度三个竞争指标,算力规模已成为 AI 领域的核心竞争资源,Meta 2024 年宣布算力组合相当于 600,000 张 H100, xAI 2025 年将 Colossus 扩展至 200,000 张 GPU。 集群从千卡扩展到万卡不是简单乘以 10,而是各维度同时质变,对比如表18-2所示。 表18-2 千卡到万卡质变对比 维度 千卡级 (约1,024 GPU) 万卡级 (约10,000+ GPU) 十万卡级 (约100,000+ GPU) 物理拓扑 单层 Spine-Leaf 即可 需要 3 层 CLOS 或 DragonFly+ 需要跨 Pod/跨建筑互联 网络复杂度 256 条链路 约10,000 条链路 >100,000 条链路 MTBF(故障间隔) 1 次 / 数天 1 次 / 2-3 小时 1 次 / <1 小时 调度延迟 秒级 秒到分钟级 需要分层调度 存储 I/O 约100 GB/s 聚合带宽 约1 TB/s 聚合带宽 约10+ TB/s 聚合带宽 监控数据量 约100 GB/天 约1-10 TB/天 约50+ TB/天 运维团队 3-5 人 10-20 人 50+ 人 规模规划应关注两条工程约束:

  1. 故障频率随规模恶化:Meta 在 Llama 3 405B 训练的 54 天中发生 466 次中断(计划内 47 次、意外 419 次),约 78% 归因硬件,有效训练时间占比仍大于 90%,全程仅需 3 次人工干预。规模越大,故障恢复与自动运维投入必须越重, 无自愈能力则难以支撑十万卡规模。
  2. 瓶颈随规模迁移:万卡集群核心瓶颈按重要性排序为网络通信(AllReduce 可占单次迭代 30-50%)、故障恢复(无自愈 时一次 GPU 故障可致数小时回滚)、数据供给(DataLoader 成为瓶颈)与调度效率(数千任务排队),后续小节围绕 这些挑战逐一展开。 集群扩展划分为三个维度:Scale-Up 用高带宽私有互联(NVLink、NVSwitch)把多卡耦合为一张“大 GPU”,用带宽换 可编程性;Scale-Out 用高速网络(InfiniBand、RoCEv2)把节点连成统一资源池,用规模换容量;Scale-Across 跨数 据中心互联。NVIDIA 各代 Scale-Up 能力持续翻倍,B200 达 1,800 GB/s,GB200 NVL72 把 72 张 Blackwell GPU 液冷互 连在一个机柜内,让 72 卡对程序员呈现为一张超级 GPU。三维度对比如表18-3所示。 表18-3 集群扩展三维度对比 扩展方式 英文名称 物理范围 互联技术 典型带宽(每GPU) 延迟 Scale-Up Vertical Scaling 单机/单柜内 NVLink + NVSwitch 9001,800 GB/s 约1 μs Scale-Out Horizontal Scaling 数据中心内 InfiniBand / RoCE 50100 GB/s 约3-10 μs Scale-Across Cross-DC Scaling 跨数据中心 Spectrum-XGS / 长距光纤 专用链路 约50 μs - 1 ms 规模规划先算“多少卡需要 NVLink 域内互访”(即 TP/SP 域),再倒推域内卡数与机柜形态,最后才决定 Scale-Out 网络 规模。

18.1.2 硬件选型

  1. 主流加速器决策矩阵 主流加速器的选型决策矩阵如表18-4所示。 表18-4 主流加速器选型决策矩阵 决策因素 H100 H200 B200 MI300X TPU v5p 训练性能 (大模型) ★★★★★ ★★★★★ ★★★★★★ ★★★★ ★★★★★ 显存容量 ★★★ ★★★★ ★★★★★ ★★★★★★ ★★★ 推理性能 ★★★★ ★★★★★ ★★★★★★ ★★★★ ★★★★ 软件生态 ★★★★★ ★★★★★ ★★★★★ ★★★ ★★★★ (JAX) 能效比 ★★★★ ★★★★ ★★★★★ ★★★★ ★★★★★★ 可获得性 ★★★ ★★★ ★★ ★★★★ ★★ (仅 GCP) 单卡成本 约$30K 约$35K 约$40K 未公开 GCP 按需计费 场景化推荐如下。 +-----------------------------------------------------------+ | Scenario Recommended Configuration | +-----------------------------------------------------------+ | Large Model Pretrain (100B+) H100/B200 + NVLink + IB | | Large Model FT (LoRA/QLoRA) H100/H200 (Large VRAM First) | | Distributed Inference H200/B200 (Large Memory+BW) | | Small Model Train (<7B) A100/L40S (Cost Priority) | | Native TPU Users (JAX) TPU v5p (Best Perf/Eff) | | Non-NVIDIA Path Explore MI300X (Cost vs Ecosystem) | | Enterprise Private Deploy DGX Appliance (Out-of-box) | | Cloud Elastic Training On-demand GPU (Karpenter) | +-----------------------------------------------------------+ 核心判断:训练是最大固定成本与时间风险,H100/B200 加 CUDA 生态短期不可替代;大显存芯片(H200、MI300X) 的价值在推理与长上下文,而非峰值算力。
  2. 中国特供 GPU 要点 出口管制催生中国市场特供 GPU,选型先判断算力与互联哪个是约束。 H800:为满足 BIS 2022 年 10 月出口管制性能密度门槛,削减 Hopper 架构的 NVLink 带宽至恰好低于阈值,其余规格 与 H100 完全一致,本质上是大致减半 NVLink 的 H100。管制生效后 H800 成为中国 AI 公司的默认训练 GPU, DeepSeek、字节跳动、阿里巴巴、百度、腾讯、商汤等头部厂商均围绕它建设训练基础设施,2023 年出现一波赶在新 规收紧前的批量囤货潮。受带宽约束,中国团队被迫转向 Expert Parallelism 与 Pipeline Parallelism,DeepSeek V3 即 在 2,048 张 H800 上完成训练,证明架构创新可部分弥补硬件限制。 H20:2023 年 10 月 BIS 改以 TPP 与 PD 为管制参数后,NVIDIA 推出“保互联、切计算、增内存”的反向路线,精确满 足 TPP 门槛(4,800)与 PD 门槛(5.92): H20 TPP Calculation (INT8, regulated precision): INT8 TOPS = 296 Bit width = 8 bit TPP = 2 x 296 x 8 = 4,736 Absolute ban threshold = 4,800 Difference = 64 (only 1.3% margin) H20 PD Calculation:
   PD = 4,736 / 814 mm^2 ~= 5.82
   Threshold = 5.92
   Judgment = Below threshold, no notification required

设计策略:保留完整 GH100 大芯片(814 mm^2)拉低 PD,禁用大量 SM(132→78)压低 TPP,保留全部 6 个 HBM3 堆叠不受内存管制。结果是算力更弱、但比 H100 多 20% 显存和高 19% 带宽的芯片,推理场景反而占优,如表18-5所 示。 表18-5 H20 与 H100 推理指标对比 指标 H100 H20 推理优势 运算/字节比(FP16) 295 37 极低,典型的内存带宽密集型 单卡可容纳 Llama 70B (8-bit) 否(需 2 卡) 是(96 GB 足够) 单卡推理,消除跨卡通信 vLLM 单卡吞吐(Llama 70B) 基准 +20% tokens/s HBM 带宽更高 DeepSeek-V3/R1 FP8 部署 需 16 卡(跨节点) 仅需 8 卡(单节点) 免除跨节点 InfiniBand 通信 对 MoE 模型的推理,Triton Fused-MoE 内核在 H20 上进一步放大优势,如表18-6所示。 表18-6 H20 上 Triton Fused-MoE 内核加速 MoE 模型 FP8 内核加速比(vs vLLM 默认) 说明 Mixtral 8x7B 最高 2.74x (batch=1) 小批量场景优势最大 DeepSeek V2 Lite (128E) 最高 2.59x (batch=32) 大专家数场景 36 种形状几何平均 1.39x 全面优于默认配置 H20 训练侧限制明显:FP16 dense 算力仅为 H100 的约 1/7,实测有效吞吐差距约 2.8 倍: H100 measured MFU ~38% (communication bottleneck during large model training) H20 measured MFU ~90% (lower compute = no communication bottleneck) Effective throughput: H20 ~133 TFLOPS vs H100 ~375 TFLOPS = ~2.8x gap 大规模预训练仍非首选,但中模型微调(LoRA/QLoRA)、RL 后训练与合成数据生成等内存敏感场景有独特优势。中国头 部厂商对 H20 的采用格局如表18-7所示,采购一度因政策收紧短期暂停、随后恢复,反映管制下算力供应格局的频繁变 动。 表18-7 中国市场的 H20 采用格局 公司 采购规模 应用场景 字节跳动 大规模囤货 豆包大模型推理,火山引擎云 阿里巴巴 大规模直接采购 通义千问推理,阿里云 AI 服务 腾讯 批量采购 元宝应用,混元模型推理 DeepSeek 确认采购方(H200 获批前用 H20) V3/R1 推理部署 H20 对集群设计的影响:

  1. 推理优先规划:TP 表现弱于 H100 但 PP、DP 不受影响,常见“H800 训练 + H20 推理”混合架构。
  2. 内存驱动批处理:96 GB 大显存允许更大 micro-batch 与更长 KV Cache,降低推理 GPU 分片需求。
  3. 算力并非总瓶颈:DeepSeek V3 在 2,048 张 H800 上训练成功,印证内存带宽与容量常比峰值算力更关键。
  4. NVLink 恢复完整:H20 恢复 900 GB/s 完整 NVLink(H800 所缺失),单节点 8 卡 TP 不再受限。 Blackwell 时代的中国特供方案如表18-8所示。 表18-8 Blackwell 中国特供方案 型号 架构 定位 B20(原计划) Blackwell 原计划面向中国市场的 Blackwell 型号,后搁置 RTX 6000D Blackwell GDDR7 显存,推理优化 B30A Blackwell (单芯片) HBM 显存,NVLink 互联,面向合规市场的中国特供方案 后续芯片均在 TPP 与 PD 阈值内设计,延续“合规优先”策略。
  1. 国产昇腾体系要点 华为昇腾是当前中国市场不可回避的选型对象,与 NVIDIA 特供 GPU 形成两条并行路线。
  1. 单芯片算力仍落后:910C 纸面规格接近 H100(约 400-640 TFLOPS FP16,不同来源估计差异较大),独立评估实测约 60% 水平,训练与推理的实测对比如表18-9所示。 表18-9 昇腾与 NVIDIA 训练性能对比 对比 华为声明 独立评估 说明 910B vs A100 80-120% A100 效率 约70-80% A100 LLM 训练 华为官方 HC 2024 910C vs H100 对飙 H100 约60% H100 性能 (DeepSeek 实测) JimmyResearch 910B 推理 (Qwen-7B) — +14.7% vs A100 (prefill) 得益于 HBM 带宽 910B 推理 (SDXL) — -38.6% vs A100 扩散模型内核优化不足 910B 超长上下文 (128K) — p99 延迟 -36% vs A100 内存子系统更稳定
  2. 系统级扩展补足差距:Atlas 900 A3 SuperPoD 以 48 节点 384 卡约 300 PFLOPS FP16 交付,CloudMatrix384 把 384 张 NPU 呈现为单台逻辑机器,复刻 NVL72 的 Scale-Up 思路,规格如表18-10所示。 表18-10 Atlas 900 A3 SuperPoD 规格 参数 规格 基础单元 8x Ascend 910C NPU / 节点 参数 规格 节点数 48 个计算节点 总算力 约300 PFLOPS FP16 机柜数 16 个 (12 计算 + 4 通信) 互联网络 UnifiedBus (LingQu) 1.0 + RoCE v2 拓扑 2 层 Fat-Tree (L1: 节点级 UB Switch; L2: 机柜级 UB Switch) 跨节点带宽损失 <3% 已交付数 >300 套
  3. 部署重心在中国市场:运营商与互联网大厂是主要采购通道,DeepSeek V4 原生优化昇腾、R1 推理部署成为需求锚 点,安可认证强制政府采购目录覆盖昇腾 310/910,如表18-11所示。 表18-11 昇腾中国市场部署现状 阵营 代表 规模 运营商 中国移动、电信、联通 主要采购通道:大规模集中采购 互联网 字节跳动、阿里巴巴、腾讯 大规模订单与规划采购 DeepSeek V4 原生优化昇腾, R1 推理部署 关键需求锚点 政府/金融 安可认证强制国产芯片 进度低于 30% 的数据中心项目必须移除进口芯片 安可认证 多款国产 AI 芯片获批政府采购 Ascend 310/910 均在列
  4. 网络效应替代原始性能:系统级扩展、垂直软件栈(CANN → torch_npu → vLLM-Ascend)、政策强制(安可认证与 国产化要求)与需求锚定四因素叠加,昇腾正成为中国 AI Infra 的事实标准。CANN 的 SIMT 兼容编程模型把 CUDA 兼 容性从附加功能变为设计要求,选型应判断 CANN 迁移成本与生态成熟度,而非单芯片算力。
  1. 服务器平台要点 服务器平台决定密度、功耗与运维复杂度,三条路线各有取舍。 •NVIDIA DGX 系列:AI 服务器的工业标准参考设计,三代规格对比如表18-12所示,机柜级供电与液冷须按功率密度预 留。 表18-12 DGX 系列规格对比 型号 GPU 配置 CPU 系统内存 网络 存储 最大功 机柜空 耗 间 DGX 8x A100 2x AMD EPYC 7742 (128 2 TB 8x ConnectX-6 30 TB 6.5 kW 6U A100 80GB 核) 200Gb/s NVMe DGX 8x H100 2x Intel Xeon 8480C (112 2 TB 8x ConnectX-7 30 TB 10.2 8U H100 80GB 核) 400Gb/s NVMe kW DGX 8x B200 2x Intel Xeon 8570 (112 480 GB 8x ConnectX-8 30 TB 14.3 10U B200 180GB 核) DDR5 800Gb/s NVMe kW •Meta Grand Teton:OCP 标准自研路线代表,基于 OCP Open Rack v3(19/21 英寸机架可互换),GPU 与 CPU 模块 分离可独立升级,每 GPU 独立网络路径避免 PCIe 交换瓶颈,统一供电散热消除节点独立 PSU 与风扇。设计已开源, 任何厂商可基于 OCP 规范生产兼容硬件,Llama 3 405B 即在此平台训练。

18.1.3 网络选型

  1. 网络是万卡集群瓶颈 千卡级数据并行中 AllReduce 梯度同步可占单次迭代总时间的 30-50%,且为全局同步操作,最慢链路决定整体性能,一 个节点网络退化会拖停整个任务。网络带宽翻倍通常带来 15-25% 有效训练速度提升,省网络就是省 GPU。与传统数据 中心不同,AI 训练流量以少数大象流、周期性爆发为主,峰值带宽利用率 80-95%,对微秒级抖动高度敏感,须针对高并 发、低拥塞的集合通信模式专门设计。
  2. 并行策略的带宽分层 通信带宽决定并行策略的使用域:TP 每层频繁同步、对延迟最敏感,必须限定在 NVLink 域内;DP 通信量大但频率低, 适合跨节点网络;PP 点对点通信要求中等延迟;EP(MoE 模式)带宽需求低,可跨越更大物理范围,如图18-1所示。 Bandwidth (per GPU) NVLink Domain InfiniBand/RoCE Domain Cross-Rack/Pod 1,800 GB/s ─┐ │ Tensor Parallel (TP) 900 GB/s ─┤ Sequence Parallel (SP) │ 100 GB/s ─┼────────────────────── │ Pipeline Parallel (PP) 50 GB/s ─┤ Data Parallel (DP) │ 25 GB/s ─┼──────────────────────────────────────────── │ Expert Parallel (EP) 10 GB/s ─┤ Data Parallel (Large Cluster) │ 图18-1 通信带宽需求与并行策略的分层对应
  3. 物理拓扑选型 大型 GPU 集群物理拓扑主要有三种:Fat-Tree(CLOS)适用性最广、是中小规模与云厂商默认选择,但交换机随规模线 性增长、成本高;Rail-Optimized 让每轨 GPU 构成无竞争通信路径,DGX SuperPOD 与 Meta Llama 3 均采用; DragonFly+ 组内全互联、组间稀疏全局链路,十万卡以上成本效益更好。拓扑选型结论如表18-13所示。 表18-13 集群物理拓扑选型 集群规模 推荐拓扑 跳数 成本指数 <1,024 GPU 2 层 Fat-Tree / Spine-Leaf 2 1x 1,024~10,000 GPU 3 层 CLOS 或 Rail-Optimized 3-5 5-8x

10,000 GPU Rail-Optimized + 多平面,或 DragonFly+ 3 10-20x

  1. Spectrum-X 以太网方案 Spectrum-X 让以太网逼近 InfiniBand 有效吞吐,工程价值是不需自建 IB 运维能力即可获得端到端拥塞规避与性能隔 离。xAI Colossus 是最大部署:200,000 GPU,宣称有效吞吐率 95%(传统以太网约 60%)。分档选型:Spectrum-X 面 向数据中心内(单平面约 2K GPU、多平面 128K GPU);跨数据中心用 Spectrum-XGS(NCCL 跨区域性能提升约 1.9x);下一代 Photonics 共封装光学每 1.6Tb/s 端口功耗降约 5x,面向超大规模规划。
  2. IB 与 RoCEv2 决策 选型阈值:超过 1 万卡的训练集群强推 InfiniBand;数百至数千卡与推理场景 RoCEv2 更优。决策树如下。 Need InfiniBand? │ ├─ Single cluster size > 10,000 GPUs? ────▶ YES ▶ Strongly recommend IB (or Spectrum-X) │ ├─ Have professional IB ops team? ────────▶ NO ▶ Consider RoCEv2 + Spectrum-X │ ├─ Budget allows 5-10x network cost? ──▶ NO ▶ Consider RoCEv2 │ ├─ Need multi-vendor ecosystem flexibility? ────▶ YES ▶ Consider RoCEv2 (wait for UEC maturity) │ ├─ Need ultra-low latency (<1μs)? ────▶ YES ▶ Choose InfiniBand │ └─ NVIDIA full-stack user? ──▶ YES ▶ Recommend IB (end-to-end optimization) RoCEv2 在 Meta Llama 3 上支撑了 16K GPU 规模训练,但需专项网络工程投入;不具备该工程能力时,万卡以上仍应优 先 InfiniBand。开放标准 UEC(Ultra Ethernet Consortium)目标是以开放的以太网通信栈替代 IB 独占地位,UALink 与 NVLink 竞争加速器互联,两者均处生态早期,短期内不改变选型结论。

18.2 平台与调度实战

18.2.1 调度层角色定位

平台与调度层是千卡/万卡集群的“操作系统”,需要回答三个核心命题:资源分配(哪个任务得到 GPU、多少张、何 时)、放置策略(GPU 落在哪些物理节点、如何保证通信拓扑最优)、公平共享(数千研究员与数百项目之间如何公平分 配算力)。这些命题的难度随集群规模呈指数级上升。 业界存在两条主流调度路径,如表18-14所示。 表18-14 Kubernetes 与 Slurm 两条调度路径 维度 Kubernetes 扩展路径 HPC Slurm 路径 代表技术 Volcano, Kueue, KAI Scheduler, KubeRay Slurm, Slurm on K8s (SUNK), Slinky 设计哲学 云原生、弹性、声明式 HPC 验证、作业管理、命令式 核心优势 极致弹性、丰富生态、自愈能力 拓扑感知、大规模作业管理、公平共享 适合场景 多租户动态共享、微服务+训练混合 离线大规模训练、HPC 工作负载 代表用户 Google, Spotify, Uber CoreWeave, NVIDIA, 多数 TOP500 超算 第三条道路是 Hybrid 混部:通过 Slurm on Kubernetes (SUNK) 或 NVIDIA slurm-operator 将 Slurm 的成熟调度能力搬 上 K8s,兼顾两类优势。

18.2.2 Kubernetes 生态

传统 HPC 社区质疑 K8s 能否胜任大规模 GPU 调度,但 Google、CoreWeave 等先锋的实践证明,经过深度定制的 K8s 可以驾驭数万张 GPU。决策视角看,K8s 在 AI 场景的核心价值如表18-15所示。 表18-15 K8s 原生能力在 AI 训练中的价值 K8s 原生能力 AI 训练场景中的价值 声明式 API 以 YAML 描述训练任务,版本控制和 GitOps 友好 Pod 抽象 每个 GPU 进程封装为 Pod,统一管理生命周期 服务发现 训练节点自动发现彼此,无需手动配置 rank 地址 自愈能力 GPU 故障 → Pod 自动驱逐 → 重调度到健康节点 多租户 Namespace + RBAC + ResourceQuota 实现租户隔离 生态可移植 一套配置在本地、云上、混合云之间可迁移 大规模落地时,原生 K8s 的调度器、自动扩缩器与 GPU 健康管理均需深度定制:Gang Scheduling 需替换调度器,GPU 故障需 taint-based 机制自动驱逐,InfiniBand fabric 需与 Pod 调度联动保证通信拓扑最优。Google GKE 是云原生路径 的参考:单集群支持 65,000+ 节点,用 ProvisioningRequest API 提前预留节点。

18.2.3 集群资源管理

NVIDIA GPU Operator 是管理 GPU 软件栈的事实标准,自动化了驱动、Container Toolkit、Device Plugin、DCGM Exporter、GPU Feature Discovery、MIG Manager、DRA Driver 与 NVSentinel 的生命周期。在十万卡集群中,手动管 理每张 GPU 的驱动版本、运行时配置与设备插件不现实,GPU Operator 将这一切变为声明式配置并自动维护一致性。 驱动升级策略上,建议在维护窗口按节点滚动升级,并配合容器化驱动的镜像切换实现快速回滚。 Karpenter 是 AWS 开源的下一代节点自动扩缩器,相比 cluster-autoscaler 在 GPU 场景优势显著,对比如表18-16所 示。 表18-16 Cluster Autoscaler 与 Karpenter 对比 特性 Cluster Autoscaler Karpenter 扩缩速度 分钟级(通过 Node Group) 秒级(直接创建实例) 实例类型选择 预定义 Node Group,受限于几种类型 动态评估数百种 GPU 实例 GPU 感知 基础支持 原生支持 nvidia.com/gpu 等标签 碎片整理 无 Consolidation 自动合并低利用率节点 Karpenter 通过 NodePool 声明式描述 GPU 需求:requirements 指定 GPU 厂商(nvidia)、型号(h100)与数量下 限,taint 限定仅 GPU 工作负载可调度,limits 控制集群 GPU 总量上限。Consolidation 在节点利用率不足时自动合并实 例,减少碎片。

18.2.4 Rendezvous 机制

分布式训练所有进程需先完成 Rendezvous(集合),彼此发现并建立通信通道,才能开始训练。传统的全互联 Rendezvous 复杂度为 O(N ):N = 1,024 时约 100 万条连接、耗时数分钟,N = 16,384 时约 2.68 亿条连接,可能直接 崩溃。标准解法是分层/树形初始化:先在小范围内完成组内 KV 交换,再由组长完成组间全局拓扑交换,最后广播回所有 成员,总连接数降为 O(N log N )。Meta 在 Llama 3 405B 训练中通过优化 Rendezvous,将组初始化时间从数小时缩短 到数分钟。PyTorch 的 torch.distributed.init_process_group 已内置分层 Rendezvous 支持。

18.2.5 调度层技术选型决策

不同规模与背景团队的调度方案选型如表18-17所示。 表18-17 调度层技术选型决策 你的场景 推荐方案 理由 创业公司,< 256 GPU K8s + Kueue + Karpenter 开源、轻量、生态好 中等规模,256-2,000 GPU K8s + Volcano + GPU Operator Gang Scheduling 成熟 大型企业,2,000-10,000 GPU K8s + KAI Scheduler / Slurm on K8s 拓扑感知 + 层级调度 极致规模,>10,000 GPU Slurm on K8s (SUNK) HPC 验证的调度能力 NVIDIA 全栈用户 Base Command + KAI Scheduler 端到端集成 HPC 背景团队 Slurm + slurm-operator 团队熟悉度优先 参考技术栈:Argo Workflows 编排工作流,KubeRay 管理分布式计算,Volcano 或 KAI Scheduler 作批调度,Kueue 实 现多租户队列,GPU Operator 与 HAMi 分管 GPU 管理与虚拟化,Karpenter 负责弹性扩缩,Multus CNI 与 InfiniBand Plugin 打通网络,CSI Driver 挂载 Lustre/NFS 存储,DCGM Exporter 与 Prometheus 负责监控,Rancher 提供多集群管 理界面。

18.3 训练与数据工程实战

本节从集群实践视角聚焦训练框架选型与数据存储决策。18.3.1 对比主流训练框架,18.3.2 与 18.3.3 剖析 Llama 3 405B 与 DeepSeek-V3 两个代表性案例,18.3.4 至 18.3.7 覆盖 GPUDirect Storage、行业存储架构、性能基准测试与决策建 议,18.3.8 展望训练框架未来方向。

18.3.1 框架生态系统对比

主流框架在灵活性、性能、生态集成与学习曲线之间各有取舍,截至 2025 年的主要选项如表18-18所示。 表18-18 分布式训练框架对比 特性 Megatron-LM DeepSpeed FSDP(PyTorch) JAX Ray Train 类型 研究代码库 PyTorch 库 PyTorch 核心 编译器框架 分布式启动 器 3D 并行 原生(最优) 通过 Megatron 集成 通过 PyTorch API 编译器推导 仅编排 专家并行 支持(MoE) 通过 Megatron 手动集合操作 自动 不支持 异步检查 支持 支持(ZeRO) 支持(DCP 异步) 支持(Orbax) 不支持 点 MFU 记 52%(1T,3072 44%(176B,800 38-43%(405B,16384 50-62%(540B,2048 N/A 录 A100) V100) H100) TPUv4) 最适合 前沿规模,100B+ 参数 多用途中等规模 PyTorch 原生,中等规模 Google/TPU 生态 多框架编排 选型决策:10B 以下标准架构用 FSDP + torch.compile,自定义架构用 DeepSpeed;10B-100B 偏 PyTorch 生态用 DeepSpeed ZeRO + Megatron TP,偏 JAX/TPU 用 GSPMD 自动分片;100B 以上前沿规模用 Megatron-LM(最大控制 力),标准 decoder-only 架构可用 TorchTitan。

18.3.2 案例研究 Llama 3 405B

Llama 3 405B(arXiv:2407.21783)由 Meta 于 2024 年 7 月发布,在 16,384 张 H100 上训练,是文档最完备的大规模训 练案例。两阶段 4D 并行分解如表18-19所示。

  1. 并行配置 表18-19 Llama 3 405B 并行分解(16K H100) 维度 8K 阶段 128K 阶段 描述 TP(张量) 8 8 节点内,NVSwitch CP(上下文) 1 16 长上下文序列级并行,128K 阶段启用 PP(流水线) 16 16 16 个连续节点,1F1B 调度 DP(数据) 128 8 流水线副本数,梯度同步维度 两个阶段都满足 16,384 = DP × CP × TP × PP,每个流水线副本为 16 节点 × 8 GPU 共 128 块,持有一份完整模型副 本。
  2. 网络基础设施 集群用 RoCEv2(每节点 8 × 400 Gbps,Clos/胖树拓扑)而非 InfiniBand,由 Meta 的以太网运维经验与规模成本驱 动。自定义拥塞控制算法至关重要:单条拥塞链路会让全部 16,384 块 GPU 在等待梯度 All-reduce 时停顿,吞吐量崩 溃。
  3. 训练效率 MFU 集群 MFU 约 38-43%,低于 Megatron-LM 在 A100 上的 52% 记录,原因包括早期 H100 FP8 内核不成熟、16K 规模下 RoCEv2 引入更多尾延迟、训练稳定性的可靠性开销。54 天完成 15.6 万亿 token 训练(16,384 张 H100-80GB),有效吞 吐约 600 TFLOPs/s/GPU。
  4. 经验总结
  1. 可靠性工程与性能工程同等重要:16K 规模下硬件故障是常态,自动节点排空与重新集成、NCCL 看门狗、基于健康检 查的抢占是标配。
  2. 异步检查点不可或缺:分层 NVMe 检查点把有效保存时间压到分钟级,多次中断的算力损失从同步保存的数十小时降 至数小时。
  3. RoCEv2 在大规模下可行:靠自定义拥塞控制,但需专项网络工程投入。
  4. FSDP 已达生产就绪成熟度,无需 Megatron-LM 代码库的复杂度即可训练顶级开放模型。

18.3.3 DeepSeek HAI-LLM

DeepSeek-V3(arXiv:2412.19437)是 671B 参数、每 token 激活 37B 参数的混合专家模型,仅在 2,048 张 H800 上完成 训练。H800 的 NVLink 带宽被出口管制降至约 400 GB/s,DeepSeek 将限制转化为架构创新催化剂,自研 HAI-LLM 框 架。

  1. 工程约束与成本口径 预训练约 14.8 万亿 token 耗时约 2 个月,消耗 2,788,000 张 H800 GPU 小时,按约 $2/GPU 小时估算成本 557.6 万美 元。关键约束是受限 NVLink 带宽:在 H100 上近乎免费的节点内 All-reduce,在 H800 上成为瓶颈,应对是彻底消除张 量并行。
  2. 并行配置与路由约束 采用 PP=16、EP=64、ZeRO-1 DP、无 TP 的配置。节点限制路由让每个 token 最多分发到 4 个节点(平均每节点最多 3.2 个专家),确保 All-to-all 通信不超过节点内 NVLink 带宽,避免节点间网络饱和。
  3. HAI-LLM 实现增量 核心增量包括:面向 All-to-all、FP8 GEMM 与流水线调度的自定义 CUDA 内核;PTX 级编程把 132 个 SM 中的 20 个专用 于通信,通信完全隐藏在计算中;无辅助损失负载均衡消除 token 丢弃。PTX 级编程带来 15-20% 吞吐量提升,代价是开 发复杂性。
  4. 训练结果 训练过程损失突增显著减少,无需回滚重启,归因于无辅助损失负载均衡、FP8 细粒度量化与端到端数值控制;多 token 预测(MTP)辅助目标提高样本效率,额外计算成本微乎其微。
  5. 对比 Megatron/DeepSpeed 三种框架的实现差异如表18-20所示。 表18-20 训练框架方法对比 方面 Megatron-LM/NVIDIA DeepSpeed/Microsoft DeepSeek HAI-LLM 主要并行策略 节点内 TP,跨节点 PP+DP 跨节点 ZeRO-3 DP DualPipe PP + EP,无 TP 节点内假设 NVSwitch 900 GB/s NVLink 900 GB/s(通用) NVLink 400 GB/s(受限) 通信重叠 梯度同步与反向重叠 All-gather 与反向重叠 专用 SM 负责通信,完全重叠 流水线调度 交错 1F1B PipeDream-Flush(1F1B) DualPipe(双向) 方面 Megatron-LM/NVIDIA DeepSpeed/Microsoft DeepSeek HAI-LLM TP 消除 必需(≤8 GPU) 可选 DualPipe 使能,关键成本节省者 验证规模 1T MoE,3,072 A100 176B 密集,800 V100 671B MoE,2,048 H800
  6. 核心经验
  1. 约束催生创新:受限 NVLink 带宽推动 DualPipe 通信-计算重叠、Warp 专化与消除 TP 的决策。
  2. 针对特定硬件积极优化:HAI-LLM 专为 H800 的 SM 数量、NVLink 带宽与内存层次优化,挖出通用框架留在桌上的效 率。
  3. 无辅助损失负载均衡在大规模下可行,这一发现对所有 MoE 架构都有意义。

18.3.4 GPUDirect Storage

GPUDirect Storage(GDS)使 NVMe SSD 与 GPU 内存之间直接 DMA 传输,绕过 CPU 与系统内存,消除传统 I/O 路径 的中间复制。对检查点与数据加载两个存储密集操作提升显著:700 GB 检查点写入从约 40 分钟降至约 1.7 分钟(约 24x),随机读数据加载从 24 GB/s 提升到 56 GB/s(约 2.3x),I/O 期间 CPU 利用率从 100% 降至 5% 以下。

  1. 部署清单 硬件要求: •NVIDIA GPU:A100、H100 或 L40S(Ampere 架构起支持 GDS) •支持 PCIe 点对点的 NVMe 驱动器(Samsung PM9A3、Kioxia CM6、Intel/Solidigm P5520 等) •服务器 BIOS 需禁用或配置 PCIe ACS 为 P2P——多数服务器默认禁用,必须显式启用 软件栈: •NVIDIA GPU 驱动 ≥ 470.57.02(A100)或 ≥ 525.60.13(H100) • nvidia-fs 内核模块(开源,提供 GDS 文件系统支持) • cufile 用户空间库(CUDA Toolkit ≥ 11.4 的一部分) •支持 GDS 的文件系统(WEKA、DDN Lustre+IME、VAST Data,或本地 ext4/XFS) 验证:

18.3.5 行业实践

 # Check GDS status and NVMe compatibility
 nvidia-smi -q | grep -A 5 "GPUDirect Storage"
 /usr/local/cuda/gds/tools/gdscheck -p
  1. Meta Llama 3 训练基础设施 Meta 的 Llama 3 训练基础设施是超大规模 AI 存储实践的最大公开披露:16,384 张 H100 GPU 分布在两个集群,定制 Grand Teton 平台,RoCEv2 400G 互连(每 GPU 400 Gbps)。其存储分层如表18-21所示。 表18-21 Meta Llama 3 存储分层 层级 技术 容量 用途 本地 NVMe 定制 Grand Teton NVMe 阵列 每节点约 30 TB 检查点暂存,数据缓存 热层(检查点) Tectonic(Meta 内部分布式 FS) 多 PB 持久检查点存储,完整 405B 检查点约 6 分钟 温层(数据集) Hammerspace 并行 NFS >50 PB 训练数据,评估数据集 冷层(归档) 未公开披露 EB 级 历史检查点,旧数据集 Meta 通过三步把检查点写入时间从 40 分钟降至 6 分钟:先写本地 NVMe,再异步上传 Tectonic(持久、纠删码),最后 经 Hammerspace 暴露用于遗留访问。
  2. NVIDIA Eos Eos 的存储规格如表18-22所示。 表18-22 NVIDIA Eos 存储规格 规格 详细信息 GPU 算力 4,608 张 H100 GPU(576 个 DGX H100 节点) 互连 InfiniBand NDR400(400 Gbps) 存储 DDN EXAScaler Lustre 配 DDN SFA NVMe 设备 峰值存储带宽 约1 TB/s(聚合) 元数据性能 >100 万元数据操作/秒 Eos 的 DDN EXAScaler 配置三级:SFA NVMe 全闪存热池承载主动训练数据与检查点,HDD 温池归档实验,IME(无限 内存引擎)突发缓冲区在降级到 Lustre 前吸收检查点写入。
  3. Microsoft Azure Azure 的存储分层如表18-23所示。 表18-23 Azure 存储分层 层级 Azure 服务 特性 计算 Azure ND H100 v5 VM(每 VM 8× H100) InfiniBand NDR400 热层存储 Azure Managed Lustre 预配置:每 TB 2.5 GB/s,最高 1.2 TB/s 数据湖 Azure Blob Storage 热/冷/归档层,地理冗余 元数据 Azure NetApp Files 元数据密集型工作负载的高性能 NFS 编排 Azure CycleCloud Slurm → VM 扩缩 + 文件系统生命周期
  4. xAI Colossus Colossus 的存储架构规格如表18-24所示。 表18-24 xAI Colossus 存储架构 规格 详细信息 GPU 数量 10 万张 H100(第一阶段),扩展至 20 万张 互连 NVIDIA Spectrum-X 以太网(400G)配 GPUDirect RDMA 存储 GPUDirect Storage(GDS)+ GPUDirect RDMA 架构 全连接 GPU 架构,无传统并行文件系统瓶颈 Colossus 让数据从 NVMe 直流到 GPU HBM3,再到网络、再到对等 GPU HBM3,完全绕过 CPU 内存,把集群视为单一 的大规模 GPU 内存空间,存储只是另一个 RDMA 对等体。此规模下存储与网络界限开始模糊,数据加载成为 GPU 间 RDMA 操作。该架构仍处实验阶段,但指向极端规模 AI 基础设施的未来。

18.3.6 存储性能基准测试

  1. FIO FIO(灵活 I/O 测试器)用于原始存储性能表征,需模拟顺序读、顺序写、随机读与元数据操作四类负载:顺序读测数据 加载带宽,顺序写测检查点吞吐,随机读测 tokenized 样本的小记录访问,元数据操作测百万级文件集的枚举与分片创 建。关键指标与目标值如表18-25所示。 表18-25 FIO 基准测试关键指标 指标 目标(AI 工作负载) 含义 顺序读带宽 每 GPU 节点 >5 GB/s 数据加载能跟得上 顺序写带宽 每 GPU 节点 >10 GB/s(配 GDS) 检查点不会停顿训练 随机读 IOPS 每节点 >10 万 IOPS 小记录访问不会造成瓶颈 元数据创建/秒 >1 万 creates/秒 数据集分片创建不超时 元数据 Stat/秒 >5 万 stats/秒 数据集枚举速度快 p99 延迟(读) <10 毫秒 异常值不会停顿 GPU 流水线 p99.9 延迟(读) <50 毫秒 灾难性异常值极少出现
  2. DLIO DLIO(深度学习 I/O 基准测试)由劳伦斯利弗莫尔国家实验室与 NVIDIA 合作开发,专为模拟 AI 训练 I/O 模式构建,弥补 FIO 访问模式与 PyTorch 数据访问差异过大的缺陷。它测量多客户端协调 I/O(模拟批次开始的 I/O 风暴)、文件内半随机 访问、计算交织节奏与元数据压力。

Install and run DLIO with realistic training I/O

pip install dlio dlio_benchmark \

     ++workload.workflow=training \
     ++workload.read_type=random \
     ++workload.computation_time=0.1 \
     ++reader.data_loader=torch \
     ++reader.prefetch_size=2
  1. 关键性能指标汇总 AI 存储场景的关键性能指标定义与重要性如表18-26所示。 表18-26 AI 存储关键性能指标 指标 定义 对 AI 的重要性 IOPS 每秒输入/输出操作数 衡量处理小型随机读(tokenized 样本)的能力,不足则 GPU 停顿 吞吐量(GB/s) 每秒传输字节数 衡量流式传输大文件(检查点、epoch 数据)的能力 元数据操作/秒 文件创建/stat/open/delete 每秒次数 决定枚举数据集与打开分片的速度,百万文件集的瓶颈 延迟 p99/p99.9 99/99.9 百分位延迟 长尾延迟导致偶发 GPU 停顿,一次 p99.9 事件可消耗数分钟 流式读效率 实际带宽 / 理论带宽 低于 70% 表明存在瓶颈

18.3.7 总结与建议

  1. 存储架构决策矩阵 不同规模的存储架构选型与预算估算如表18-27所示。 表18-27 存储架构决策矩阵 规模 GPU 数 推荐架构 估算预算(存储) 小型实验室 1-8 本地 NVMe RAID-0(8× NVMe SSD) $3K-$10K 规模 GPU 数 推荐架构 估算预算(存储) 中等规模 8-64 商用 NVMe 上的 WEKA + MinIO 归档 $5万-$20万 企业级 64-512 DDN EXAScaler Lustre + CephFS 温层 $20万-$100万 超大规模 512-4K DDN/WEKA Lustre + GPUDirect Storage + Azure Managed Lustre(云) $200万-$1000万 前沿规模 4K-16K+ 定制(Meta Tectonic)或 WEKA + GDS + InfiniBand 网络 $1000万-$5000万+
  2. 存储架构师检查清单
  1. 购买前先做基准测试:用 DLIO 模拟真实训练工作负载,FIO 单独测试会产生误导。
  2. 优先保障元数据性能:确保文件系统能处理 >5 万次 stat() 操作/秒。
  3. 从第一天起使用异步检查点,同步阻塞的代价是致命的。
  4. 在任何 A100/H100 集群上启用 GPUDirect Storage,硬件已就绪,瓶颈是软件配置。
  5. 激进分层:NVMe(天)→ 并行文件系统(周)→ 对象存储(月/年),每层成本降低约 10 倍。
  6. 为数据流水线算力做规划:预处理 15 万亿 token 需要可观的 CPU 算力投入,数据就绪时点制约训练开始日期。
  7. 版本控制数据:DVC、LakeFS 或 metadata.parquet,没有数据版本控制的可复现性是幻觉。
  1. 未来展望 存储行业正由 AI 工作负载驱动代际转变:NVMe-oF 推动存储与计算独立扩展;计算存储(DPU/SmartNIC)承担更多 tokenization、解压缩与过滤;GDS 随 NVIDIA 将 GPUDirect 扩展到 DPU 而普及;云原生并行文件系统(FSx for Lustre、Azure Managed Lustre、Parallelstore)缩小与本地 Lustre/WEKA 的差距。最终方案是集成层次结构:NVMe 提供速度,并行文件系统提供规模,对象存储提供持久性,GDS 将它们绑定在一起。

18.3.8 未来方向

•预填充与解码分离:分离式服务架构让预填充与解码阶段独立扩展。 •FP4 训练:B200 原生支持 FP4,BitNet b1.58 等研究表明 4 位训练可达 Transformer 的 16 位精度。 •自动并行化:编译器自动分区是长期趋势,JAX GSPMD 最先进,PyTorch 的 tensor 与 export 正趋近类似能力。 •存储分离:权重加载成为分布式问题,vLLM、SGLang 等正开发分离式 KV 缓存与权重流式传输。

18.4 可观测性与故障实战

18.4.1 可观测性基础与三大支柱

  1. 故障是常态,不是例外 在大规模 GPU 集群中,硬件故障是日常运维常态:Meta 在 16,000 卡 H100 集群上平均每 2-3 小时发生一次需人工介入 的故障,数十万 GPU 集群每天处理数十起以上故障事件。实际故障率低于单卡 MTBF 的简单倒数推算,因为并非所有组 件失效都会导致可用性故障,ECC 可纠正单比特错误,NVLink 链路可降级运行,多数 XID 事件无需立刻停机。 更需关注故障的关联性与传播性:HGX 基板上 8 张 GPU 共享 NVLink 域,一张卡的 NVLink CRC 错误可能触发链路重训 练,导致同板其他 GPU 暂时中断通信;一个交换机端口故障会中断所有经其路由的 RDMA 流量,波及同一机架数十张 GPU。故障类别与平均修复时间如表18-28所示。 表18-28 故障类别与修复时间 故障类别 典型表现 平均修复时间 GPU 硬件 ECC 错误、XID 事件、NVLink 降级、显存故障、过热降频 数小时到数天 网络 InfiniBand/RoCE 链路中断、交换机端口故障、RDMA 超时、拓扑降级 数分钟到数小时 软件/驱动 CUDA 驱动崩溃、NCCL 超时、容器 OOM、文件系统卡顿 数分钟
  2. 不可见故障的代价 未被及时发现的 GPU 故障对训练任务的影响是灾难性的:一个慢节点会拖慢整个分布式训练的 AllReduce,使整体吞吐 下降 30-50%;计算单元产生不可检测的错误输出(SDC)导致模型收敛异常甚至发散;GPU 突然离线需从最近 checkpoint 恢复,千卡规模下重建通信拓扑可能耗时 10-30 分钟。
  3. 从监控到可观测性 监控回答系统是否正常,可观测性回答系统为什么出现这种行为。传统监控依赖已知指标与阈值假设,在 GPU 集群中并 不成立:GPU 利用率在单次迭代内剧烈波动(各阶段在近 0% 与 95% 间切换),固定阈值产生大量误报;不同模型与训 练阶段的正常基线不同。因此需要采集足够丰富、细粒度的遥测数据,在不预设故障模式的情况下对任意时间窗口、任意 维度做关联分析。

18.4.2 DCGM 与 Prometheus

现代可观测性体系建立在 Metrics(指标)、Logging(日志)、Tracing(追踪)三类数据之上(表18-29),按 Metrics → Logs → Traces 三层下钻协同:先由指标发现异常窗口,再经日志确认事件,最后用追踪定位瓶颈。 表18-29 可观测性三大支柱 支柱 回答的问题 核心组件 规模量级(万卡集群) Metric 系统多大程度接近极限 Prometheus、DCGM Exporter、 1,000 节点每 15 秒采集一次 DCGM 200+ 字段,日均 s Grafana 约 5 亿数据点 Loggi 某个具体时刻发生了什么 Grafana Loki、Fluent Bit 日均数十 TB,需分级存储与采样 ng Tracin 请求经过哪些路径、每环 Grafana Tempo、OpenTelemetry 每次迭代产生上万 Span,需采样策略 g 节耗时 Collector

  1. 核心指标解读 NVIDIA Data Center GPU Manager(DCGM)是 GPU 集群监控的官方套件,通过 NVML 直接读取 GPU 硬件寄存器,提 供远超 nvidia-smi 的 200+ 字段遥测数据。核心指标、运维含义与参考阈值如表18-30所示。SM Active 与 GPU Util 的比 值是实用的诊断信号:接近 1.0 表示计算密集;0.6-0.8 可能有内存带宽瓶颈;低于 0.5 需排查 kernel launch 开销。 表18-30 DCGM 核心指标 指标 运维含义 参考阈值 真正的计算核心活跃度,区别于 nvidia- 训练矩阵密集阶段应稳定在 85-98%;通 DCGM_FI_PROF_SM_ACTIVE smi 的时间占空比 信阶段骤降为正常 Tensor Core 活跃比例 持续低于 60% 说明大量时间花在非 DCGM_FI_PROF_PIPE_TENSOR_ACTIVE Tensor Core 运算上 DCGM_FI_PROF_DRAM_ACTIVE 显存带宽利用率 持续接近 100% 表示 HBM 带宽饱和 DCGM_FI_DEV_NVLINK_CRC_FLIT_ERRO R_COUNT_TOTAL NVLink 链路质量 累积增长预示链路降级风险 DCGM_FI_DEV_ECC_SBE_VOL_TOTAL / SBE 可纠正但指示退化;DBE 不可纠正 DBE 必致错误 DBE 触发动态页退役与节点隔离 DCGM_FI_DEV_CLOCK_THROTTLE_REASO NS 降频原因位掩码(过热、功耗、供电等) 异常降频位持续置位需排查 DCGM_FI_DEV_XID_ERRORS 驱动报告的硬件/软件错误事件 事件驱动,结合 XID 编目解读 PCIe 链路质量(金手指氧化、retimer 故 高 replay 率(>10/s)需排查 DCGM_FI_DEV_PCIE_REPLAY_COUNTER 障等)
  2. 采集开销与频率规划 各 Field Group 的采样开销与建议采集间隔如表18-31所示。经验法则:DEV 组不超过 30 秒以免漏掉瞬时掉卡事件; Profiling 字段不短于 30 秒以免采样开销影响性能;Process/Accounting 保持 60 秒以上。 表18-31 DCGM 采集开销与频率规划 Field Group 采样开销 建议采集间隔 Device (DEV) < 1µs 10-15s Profiling (PROF) 10-50µs 15-30s(数千卡同时采集会扰动性能,建议 30s) Health (HEALTH) < 1µs 15-30s Process (PROC) 50-200µs 30-60s Accounting (ACCT) 50-200µs 60-120s
  3. 诊断等级 dcgmi diag 四级诊断的耗时与适用场景如表18-32所示。Level 2 通常作为节点重新加入调度池的健康门槛;Level 3/4 用 于 RMA 决策,若 Level 3 失败(如内存测试爆出多个 DBE)即确认硬件故障并触发 RMA。 表18-32 DCGM 诊断等级 等级 命令 耗时 适用场景 Level 1 dcgmi diag -r 1 约数秒 日常快速巡检 Level 2 dcgmi diag -r 2 约2 分钟 部署前验收、节点重新入池的健康门槛 Level 3 dcgmi diag -r 3 约15 分钟 故障排查 Level 4 dcgmi diag -r 4 约45 分钟 硬件 RMA 依据
  4. 健康检查与策略引擎 dcgmi health 支持 NVLink、PCIe、内存、热、功耗逐项检查。策略引擎支持的自动动作包括 isolate、reset、page- retirement、mark-unschedulable 与 notify,可对指定 XID 事件自动执行隔离:

NVLink / PCIe / memory / thermal / power health checks

dcgmi health -s all -i 0

Policy engine: auto-isolate GPU on XID 48

dcgmi policy -g 0 --set xid,48 --action mark-unschedulable 5) 进程级作业统计 进程级统计( dcgmi stats -p )对计费和资源核算至关重要。DCGM Exporter 将 DCGM 遥测转换为 Prometheus 兼容格式,暴露在 http://:9400/metrics ,通过 CSV 配置文件控制暴露的指标(默认约 80 个字 段),K8s 中每节点一个 Pod 以 DaemonSet 部署,配官方 Grafana Dashboard(ID 12239)即构成 GPU 集群监控的事 实标准,采集架构如图18-2所示。 GPU Nodes × N GPU 0 Grafana GPU 1 :3000 dcgmd DCGM Exporter Prometheus Daemon :9400/metrics scrape /metrics TSDB Storage :9090 GPU 2 AlertManager :9093 GPU 7 图18-2 DCGM Exporter 采集架构 6) Prometheus 大规模扩展 单个 Prometheus 实例的扩展瓶颈在时间序列数量(约 10M 活跃序列)、采集吞吐(约 1M samples/s)与存储。万卡集 群的标准扩展路径是 Federation(联邦)分层拉取 + Thanos 对象存储归档 + Recording Rules 预聚合,其架构如图18-3 所示。 Cluster A (1,000 GPUs) Cluster B (1,000 GPUs) Global Layer Prometheus Server Prometheus Server Global Prometheus Scrape All Nodes in Cluste Scrape All Nodes in Cluste Federation Mode rA rB Pull Aggregated Data from Cluster Prometheus remote_write remote_write query Thanos Store & Query Long-term Storage (Object Storage MinIO / S3 Long-term Data) 图18-3 Prometheus 联邦扩展架构 Recording Rules 将高频原始指标预聚合为 cluster:gpu_util:avg 、ECC SBE 速率等长期查询友好的派生指标,配合 scrape_interval: 15s 、 retention.time=30d 与 external_labels 集群标记。 OpenTelemetry(OTel)是 CNCF 的统一遥测数据标准,解决 GPU 集群可观测性的厂商锁定问题。生产部署采用 Agent → Kafka → Gateway 三层架构:每节点一个 Agent 做轻量处理,Gateway 集群级聚合,Kafka 作中间缓冲层,同时吸收 故障时的遥测数据尖峰并避免下游后端短暂不可用导致丢数据。DCGM Exporter 原生输出 Prometheus 格式,通过 OTel Collector 的 Prometheus Receiver 可无缝接入 OTel 管道。Gateway 核心配置覆盖 Metrics + Logs + Traces 三条管线: receivers: otlp: protocols: grpc: endpoint: "0.0.0.0:4317" kafka: brokers: ["kafka-0:9092", "kafka-1:9092", "kafka-2:9092"] processors: batch: send_batch_size: 1000 memory_limiter: limit_mib: 4096 service: pipelines: metrics: {receivers: [otlp, kafka], processors: [memory_limiter, batch], exporters: [prometheusremotewrite]} logs: {receivers: [otlp, kafka], processors: [memory_limiter, batch], exporters: [loki]} traces: {receivers: [otlp, kafka], processors: [memory_limiter, batch], exporters: [otlp/tempo]}

18.4.3 故障检测与处理

  1. XID 错误编目 XID 是 NVIDIA 驱动报告错误的编码系统,当前定义 154+ 种类型。运维需要熟记的关键几类如表18-33所示。 表18-33 XID 错误编目 XID 名称 严重级别 处理建议 13 Graphics Engine Exception Warning 重启进程即可 31 GPU Memory Page Fault Warning 检查 CUDA kernel 越界访问 45 Preemptive Channel Removal Error 通常由 ECC DBE 触发,检查显存 48 Double Bit ECC Error Critical 立即隔离节点,检查是否需要硬件更换 62/63 ECC Page Retirement 失败/过多 Error 显存深度故障,需更换 GPU 74 NVLink Error Critical 检查 NVLink 线缆/交换芯片 79 GPU Has Fallen Off The Bus Critical PCIe 连接物理故障 92 High SBE Error Rate Warning 计划性维护窗口更换 94/95 Uncontained / Uncorrectable ECC Critical 可能已产生错误数据,验证 checkpoint 完整性 119/120 GPU Recovery Action Warning/Info 驱动层恢复,无需干预
  2. ECC 与动态页退役 SBE(单比特错误)由 ECC 自动纠正、不影响结果,但频繁出现是显存退化的早期信号;DBE(双比特错误)无法纠正、 必然导致结果错误,必须立即隔离该 GPU 并验证输出。从 A100 起,GPU 检测到显存页面反复 ECC 错误会自动标记退役 (单页 SBE 超阈值默认 60 次或一次 DBE),退役页数每周增加 50+ 说明显存加速退化,需计划更换,用 nvidia-smi - q -d RETIRED_PAGES 查询退役页。
  3. 静默数据损坏检测 SDC 发生在计算单元内部(ALU、Tensor Core),产生错误却看似合理的输出,无任何硬件错误报告,是大规模集群最隐 蔽的故障模式。检测手段如表18-34所示。 表18-34 SDC 检测方法对比 方法 原理 适用场景 局限性 DCGM memtest (Level 4) 写入已知模式 → 读出比较 部署前验收 只能检测显存错误 应用级 Checksum 关键张量 CRC/MD5 比对 训练过程中 增加计算开销 同构冗余计算 同一计算在两 GPU 上比对 部署前验证 资源开销翻倍 算法级检测(如 ABFT) 数值方法内建错误检测 矩阵运算 需修改训练代码
  4. NVSentinel NVSentinel 是 NVIDIA 面向 K8s 的 GPU 故障自动检测与恢复系统,以 DaemonSet 部署,对可恢复故障触发 GPU Reset,对不可恢复故障依次执行 Cordon、Taint 与更新 Node Condition,其故障防护流程如图18-4所示。 GPU Hardware DCGM NVSentinel DaemonSet Kubernetes API K8s Scheduler ECC DBE Event Detected XID 48 Assess Fault Severity alt [Recoverable Fault (High SBE Rate)] Trigger GPU Reset Reset Successful Mark Node Available [Unrecoverable Fault (DBE)]
  1. Cordon Node (Block New Pod Scheduling)
  2. Taint Node (Evict Existing GPU Pods)
  3. Update Node Condition (GPUProblem=True) Trigger Rescheduling GPU Hardware DCGM NVSentinel DaemonSet Kubernetes API K8s Scheduler 图18-4 NVSentinel 故障防护流程 硬件层(DCGM)与系统层(Node Exporter)属于白盒监控,训练任务的语义正确性与性能健康度需要应用层探针。训 练迭代是理想的检测粒度:千卡规模单次迭代 1-10 秒,一次迭代覆盖数据加载、GPU 计算、通信与优化器更新的完整链 路。在训练脚本中暴露 training_step_duration_seconds 、 training_dataload_duration_seconds 、 training_loss 与 training_tokens_per_second ,采集间隔 10 秒。
  1. 慢节点检测 慢节点(Straggler)是分布式训练最常见的性能问题。检测算法用 all-gather 计算全局平均 step 时间,本地 step 时间超 过均值 2 倍即告警,随后交叉查询对应时间窗口的 DCGM 指标(SM 利用率、温度、NVLink 错误)定位根因。
  2. MFU 测量 MFU(Model FLOPs Utilization)是训练效率的黄金标准。两种策略:自定义 Prometheus 指标直接上报 FLOPs(精 确,需改训练代码);从 DCGM 的 SM Active + Tensor Active + Clock 反推(无需改代码,但无法区分有效计算与空循 环)。 集群超过数千 GPU 时人工处理故障在物理上不可行,自愈系统将有经验运维的决策过程编码为自动化规则。自愈闭环分 五阶段,如图18-5所示。 Fail Success Service Restored DCGM XID/ECC Events Verify Detect App-layer Probe Timeout DCGM Diagnostic Verificati Prometheus Alert on Rule Engine Matching Pre-flight Test ML Anomaly Detection Historical Pattern Compari Gradual Pool Rejoin son K8s Cordon + Taint Diagnose Slurm Node Removal Network Route Isolation Isolate Recover GPU Reset Node Reboot Automated RMA Process 图18-5 自愈闭环五阶段
  1. 检测(Detect):硬件层 DCGM XID/ECC、指标异常、网络层端口错误率、应用层 step 时间异常、语义层 loss NaN 多 层覆盖。
  2. 诊断(Diagnose):把关联信号的时间窗口匹配到已知故障模式,DBE 直接判定显存故障,NVLink CRC 错误超 100 且 AllReduce 延迟超基线 3 倍判定 NVLink 降级。
  3. 隔离(Isolate):最小化爆炸半径,单 GPU DBE 驱逐容器,NVLink 域故障 taint 整个节点。
  4. 恢复(Recover):频繁 SBE 触发页退役,单次 DBE 做 GPU Reset 后重枚举与验证(成功率约 85%)。
  5. 验证(Verify):最易被忽视却最关键,DCGM 诊断 + 基线比对 + 渐进回池 + 24 小时增强监控窗口,没有验证环节的自 愈系统实际在制造不确定性。
  1. Meta GCM Meta 的 GPU Cluster Manager(GCM)是管理数万 GPU 生产环境的最成熟开源平台之一,核心创新是“一切围绕作 业”:可观测性必须回答“哪个作业受影响”而非仅“哪张 GPU 出问题”。其能力如表18-35所示。 表18-35 Meta GCM 能力 能力 实现方式 价值 GPU 遥测 → Slurm JobID 映射 读 Slurm cgroup,将 GPU 进程 PID → JobID GPU 指标到训练任务的精确归属 任务级诊断 GPU 错误事件与 Slurm 任务关联 回答“这次 XID 48 影响了哪个作业” 节点黑名单 自动将故障节点从 Slurm 可用列表移除 防止新任务调度到故障节点 RMA 管线 故障统计 → 自动生成 RMA 报告 → 对接供应商 加速硬件更换流程 节点黑名单的触发信号包括 ECC DBE、NVLink CRC 超阈值、XID 48/74/79/94/95,加入后通过 scontrol 将节点置为 drain ,并按故障类型自动生成 RMA 报告。
  2. AMD Primus 异构集群 AMD 提供对标 DCGM 的 Primus 系列:Primus-Lens 负责遥测采集与可视化,Primus-Bench 做部署前诊断,Primus- SaFE 集成故障检测与自动恢复。MI300X 为 Chiplet 设计,需关注 XCD 级指标(每个 XCD 有独立温度传感器与功耗域) 与两个层面的 Infinity Fabric 链路状态,建议在 OTel Collector 层统一指标格式,使 Grafana 面板可同时展示两种 GPU 的健康状态。
  3. CoreWeave 的集成告警 CoreWeave 将 W&B 实验级告警(loss 异常、梯度爆炸、吞吐下降)与 HPC 基础设施告警深度融合,检测到训练异常时 自动查询对应时间窗口的基础设施告警。告警分级路由:P1 Critical(DBE/NVLink 全断/loss NaN)走 PagerDuty 电 话,15 分钟 SLA;P2 Warning(慢节点/高温)走 Slack,1 小时;P3 Info 仅 Dashboard 标记;P4 Silent 仅记录。

18.4.4 日志与分布式追踪

  1. 数据量与分级 以 10,000 GPU 集群为例,日志来源包括 CUDA 驱动、NCCL 通信、框架日志、容器系统日志与 IB 子网管理器,合计约 12.5 TB/天,30 天保留即 375 TB,必须分级存储,如表18-36所示。XID 错误、ECC 事件、NCCL 错误等关键信号全量采 集、永不采样;DEBUG 等高产出流按节点限速(如每秒 100 条)。 表18-36 日志分级存储策略 日志级别 实例 存储介质 保留时间 压缩比 L1 热数据 最近 24 小时 Ingester 内存 + 本地 SSD 24 小时 无压缩 L2 温数据 过去 7 天详细日志 对象存储 (S3/MinIO) 7天 Zstd 约3:1 L3 冷数据 过去 30 天聚合日志 对象存储低频 30 天 Zstd + 采样 约10:1 L4 归档 30 天以上统计 对象存储归档 365 天 聚合统计
  2. Loki 与 Fluent Bit Loki 的设计原则是只索引标签(namespace、pod、node、gpu_uuid),不索引日志正文,与 Prometheus/Grafana 原 生集成,可在面板中从 Metrics 跳转到 Logs。Fluent Bit 以 DaemonSet 采集容器日志并打上 K8s 标签,经 tail 输入、 kubernetes 过滤与 loki 输出三段配置实现采集、加标签与推送。
  3. 日志-指标关联与 AIOps 单独一条“NCCL timeout”日志没有意义,但当 Loki 中同时出现 NCCL timeout 激增、同一节点 DCGM NVLink CRC 错 误同步上升、训练步时骤增三个信号,即可推断“NVLink 链路质量下降导致 NCCL 通信超时”。Anthropic 的实践表明日 志与指标的实时关联可实现问题前摄性检测。GPU 故障模式相对有限(XID/ECC 类型固定),适合建立监督学习模型自动 分类。 训练追踪回答“一次训练迭代跨越上万 GPU,哪一个环节成为瓶颈”。与 Web 服务不同,训练 Span 的上下文传播面临 特殊边界:GPU 侧执行由驱动异步管理,只能通过 CPU 侧 stream 同步点近似 Span 边界;NCCL AllReduce 需用 Flight Recorder 时间戳事后推断各阶段耗时;不同 rank 的 Span 关联需用全局 step 作为共享关联键,设置 iteration.step 属性即可按 step 聚合查询。
  4. 采样策略 每次迭代产生上万 Span(每 GPU 一个 root Span),必须智能采样,如表18-37所示。 表18-37 Span 采样策略 采样策略 说明 采样率 适用场景 Head Sampling Span 创建时随机决定 10-30% 常规监控 Tail Sampling Span 完成后决定 慢 Span 100% 精确捕获慢迭代 Error Full Sampling 含 error 的 Span 100% 采样 100%(错误) 故障排障 Tail Sampling 在 OTel Collector 中配置为 tail_sampling processor:错误 Span 全量、慢迭代(超均值 3 倍)全量、 其余按 10% 概率采样。

18.4.5 完整堆栈与工程实践

  1. NVIDIA Fleet Intelligence NVIDIA 提供的云端集群健康管理服务,将匿名化 GPU 物理指标(SN、UUID、温度、功耗、利用率、ECC 计数)上传云 端,提供跨集群基准对比与预测性 RMA(预测 30 天内可能故障的 GPU)。其价值在于群体智能,可发现单用户无法发现 的批次性硬件缺陷(如某批次 TIM 材料缺陷导致的异常温升)。不包含应用代码、模型权重或训练数据,企业可自主选择 是否上传。
  2. 实验层观测 W&B 与 MLflow W&B 是实验追踪的事实标准,承担应用层观测:实验级指标(loss、梯度范数)+ 系统指标(GPU 利用率、显存)+ 告 警(loss NaN、梯度爆炸)。其独特价值在于实验语义与基础设施指标的关联,运维在 Loki 看到“XID 48 on GPU 3”时,可在 W&B 查询该 GPU 当时在跑哪个实验、从哪个 checkpoint 恢复损失最小。MLflow 是开源替代,适合数据隐 私要求严格的企业。
  3. Kubernetes Node Problem Detector 通过 DaemonSet 检测节点问题并上报为 Node Condition:自定义插件每 30 秒检查 GPU 健康,检测到 XID 错误后更新 GPUHealthy=False ,调度器停止调度新 workload,配合 Descheduler 驱逐现有 Pod,修复后恢复 GPUHealthy=True 。
  4. 方案选型决策树 可观测性方案的整体选型决策树如图18-6所示。 GPU Cluster Scale & Sched uler Scheduler Type? Experiment Management Needed? Slurm Kubernetes Yes No GPU Count? GPU Count? W&B / MLflow Prometheus + Grafana onl y < 1000 >= 1000 < 100 100-1000 >= 1000 DCGM + Prometheus + Meta GCM + Prometheus DCGM Exporter + MEDIUM: + NVSentinel LARGE: + OTel Collector Custom Slurm Scripts + OpenTelemetry Prometheus + Grafana + Loki + Tempo + Kafka + Thanos/Cortex 图18-6 可观测性方案选型决策树 生产级可观测性堆栈的分层架构如图18-7所示。 Hardware Layer H800 GPUs × 10,000 InfiniBand NDR × 4,000 p GPFS / Lustre Storage orts Collection Layer - Node Level Fluent Bit DCGM Exporter IB Exporter Node Exporter OTel Agent Log Collector :9400 - GPU Telemetry :9310 - InfiniBand Telemet :9100 - CPU/Mem/Disk :4317 - App Traces ry Collection Layer - Cluster Level Prometheus Cluster OTel Gateway Scrape + Federation Metrics/Traces/Logs

(Kafka Buffer Layer) Storage Layer (Thanos + MinIO (Loki + S3 (Prometheus TSDB (Tempo + S3 365d Cold Data) 30d Logs) 15d Hot Data) 7d Traces) Alerting & Remediation Presentation Layer AlertManager Grafana Alert Routing Unified Dashboards NVSentinel PagerDuty / Slack / Webho GPU Auto-Remediation ok 图18-7 完整可观测性堆栈架构 5) 告警规则示例 告警规则的核心原则是每条告警必须可行动并附带 runbook。典型规则示例: rules:

  • alert: GPU_DoubleBitECCError expr: increase(DCGM_FI_DEV_ECC_DBE_VOL_TOTAL[5m]) > 0 for: 1m labels: {severity: critical, category: gpu-hardware}

... (omitted): low SM utilization, NVLink errors, slow rank alerts

  1. 数据流总结 GPU、应用与 K8s 三类数据分别经 DCGM Exporter、OTel Agent、Fluent Bit 汇入 Prometheus、Tempo、Loki,由 Grafana 统一展示,NVSentinel 与 Meta GCM 负责闭环处置。
  2. 关键参考链接 各核心组件的官方文档链接如表18-38所示。 表18-38 可观测性组件参考链接 组件 链接 NVIDIA DCGM https://docs.nvidia.com/datacenter/dcgm/latest/user-guide/index.html DCGM Exporter https://github.com/NVIDIA/dcgm-exporter NVIDIA GPU Dashboard (Grafana https://grafana.com/grafana/dashboards/12239-nvidia-dcgm-exporter-

#12239) dashboard/ Grafana Loki https://grafana.com/oss/loki/ Grafana Tempo https://grafana.com/oss/tempo/ OpenTelemetry https://opentelemetry.io/ 组件 链接 Prometheus https://prometheus.io/ NVIDIA NVSentinel https://github.com/NVIDIA/nvsentinel Weights & Biases https://wandb.ai/ MLflow https://mlflow.org/ Kubernetes Node Problem Detector https://github.com/kubernetes/node-problem-detector 大规模 GPU 集群的可观测性是决定训练效率与硬件投资回报率的核心基础设施,建设遵循几条原则:故障是常态,可观 测性必须是一等基础设施;三大支柱通过 Grafana 统一面板实现 Metrics → Logs → Traces 三层下钻;DCGM 是 GPU 硬 件遥测的基石;自愈闭环(检测→诊断→隔离→恢复→验证)是规模化运维的唯一路径;应用层探针把训练迭代作为最小 检测单元;规模决定选型,选型决策见图18-6。 从零搭建的分阶段路线:基础可见性 → 故障告警 → 日志聚合 → 应用层探针 → 分布式追踪 → 自愈自动化 → 大规模优 化。核心原则:先看见再理解,按需采集避免数据沼泽,每条告警可行动并附带 runbook,自愈有边界,DBE 可自动隔 离,但自动 RMA 需人工审批。

18.4.6 问题定位方法论

大规模集群的复杂性能问题常是应用、库、驱动、固件、硬件多层交织的产物。定位方法论的核心是「分层隔离 + 证据链 收集」,把模糊问题逐步收敛到确定层。

  1. 分层隔离模型 把系统按栈划分为五层,每层有明确的观测信号,如表18-39所示。定位策略是「自顶向下快速分层」:先确认问题发生在 哪一层,再在该层深入,每层排除后向下一层收敛。 表18-39 分层隔离观测信号 层 观测信号 典型问题 应用层 step 时间、loss、日志 算法错误、数据加载瓶颈 库层 NCCL 日志、cuBLAS/cuDNN 版本 算子回退、库版本不兼容 驱动层 dmesg、Xid 错误、nvidia-smi 驱动 bug、CUDA context 泄漏 固件层 BMC 日志、NVML 固件版本 固件 bug、GPU/NIC 固件不匹配 硬件层 ECC、温度、功耗、链路 CRC 物理故障、降级
  2. 证据链收集 问题发生时第一时间收集跨层证据,错过时序窗口会导致不可复现: •时间对齐:所有日志统一时间戳(NTP 同步 + 应用打点),跨层关联以时间为锚。 •分层快照:故障瞬间同时抓应用日志、NCCL 调试日志、dmesg、 nvidia-smi (含 Xid 计数)、BMC 传感器、交换机 端口错误计数。 •慢速缓存:高频指标(DCGM 15 秒级)与低频日志(秒级)并存,回溯时用高频指标缩小到秒级再翻日志。
  3. 跨层信号组合判定 单一信号不可靠,需组合判定。典型模式:DBE + 训练 OOM 超时 + NCCL 超时 → 显存物理故障;NVLink CRC 升高 + AllReduce 延迟升高 + 温度正常 → NVLink 链路降级;SM Active 低 + GPU Util 高 + 温度正常 → kernel 启动开销或同步 等待。
  4. 信息模糊场景的收敛方法 软硬件问题常信息不全(无明确错误码、偶发、不可复现)。收敛手段:隔离变量(换节点、换卡、换驱动版本,一次只 变一个变量);最小复现(把生产负载缩减为单卡单算子脚本);对比基线(同型号无故障节点做 A/B 对比,差异即线 索);历史模式(检索历史同类故障,匹配已知模式与修复动作)。
  5. 输出可落地方案 定位到根因后,输出要可落地:给出修复动作(驱动升级、参数调整、硬件替换)、验证方法(复跑最小复现 + 回归测 试)、预防措施(监控指标、版本基线、验收规范),并沉淀进故障数据库,形成“定位→修复→验证→沉淀”闭环。

18.5 可视化与平台集成实战

18.5.1 碎片化工具之痛

  1. 典型 AI 平台的工具矩阵 在一个具备完整能力的千卡级 GPU 集群中,平台团队通常会部署以下工具栈,每一层都自带独立的 Web 控制台。一个典 型的 H800 集群日常运维中,AI 工程师和数据工程师需要频繁在以下至少 5~7 个互不相通的 Web UI 之间切换,如表18- 40所示。 表18-40 典型 AI 平台工具矩阵 序 工具 Web 控制台 典型问题/操作 认证方式 号 1 Kubernetes (Rancher/RKE2) Rancher UI :8443 节点状态、Pod 列表、资源配额 Rancher RBAC / Keycloak 2 调度器 (Volcano/Kueue) Volcano Dashboard / kubectl 命令行 队列积压、任务优先级、Gang Scheduling K8s RBAC (kubeconfig) 3 训练平台 AI) (Determined Det WebUI :8080/det/ 实验列表、Trial 详情、超参对比、 TensorBoard Det Bearer Token 4 数据平台 台) (自研/内部平 Next.js 管理端 :3000 数据集版本、预处理任务、标注进度 平台自有 JWT 5 监控 (Grafana + Prometheus) Grafana :3030 GPU 利用率、NVLink 带宽、IB 流 量、节点温度 Grafana OAuth / 匿 名 6 日志 (Loki / ELK / ClickHouse) Grafana Loki / Kibana 训练日志、系统日志、错误栈 同监控体系 7 告警 (Alertmanager) Alertmanager UI :9093 告警规则、静默设置、通知路由 无/Basic Auth
  2. 转椅集成问题 业界将这种需要人工在不同系统间切换、复制粘贴信息来完成一个完整工作流的模式称为 “Swivel Chair Integration”(转椅集成)——运维人员像坐在转椅上一样,在多个显示器、多个 Web 标签页之间来回旋转。 一个典型的故障排查场景(真实时间线): 09:15 Grafana alert: node xl003 GPU 0 utilization drops from 98% to 12% 09:16 Open Rancher UI ▶ Find pods running on xl003 09:18 Open Det WebUI ▶ Search for corresponding Experiment ID, view Trial logs 09:20 SSH into xl003 ▶ nvidia-smi check GPU status 09:23 Open Prometheus ▶ Manually query xl003 DCGM XID error metrics 09:25 Open Loki ▶ Search xl003 kernel logs for ECC/NVLink errors 09:30 Comprehensive judgment: NVLink link downgrade ▶ Decide to migrate the task 09:32 Return to Rancher UI ▶ Add nodeSelector to exclude xl003 09:35 Return to Det WebUI ▶ Confirm task resumed on new node 09:40 Open Alertmanager ▶ Silence this alert for 2 hours 在这 25 分钟的排查过程中,工程师在 5 个不同工具之间切换了 9 次,多次手动复制实验 ID、节点名、时间戳等信息。对 于一个管理 200+ 节点、日均运行 50+ 训练实验的集群,这种碎片化带来的效率损失不可接受。
  3. 统一座舱需求 AI 集群运维需要的是一个 “统一座舱”(Unified Cockpit),一个单一入口、统一上下文、跨平台联动的操作界面。其核 心诉求可以归纳为三层,如图18-8所示。 Current Pain Points 5+ Separate Consoles Information silos Manual ID Copy Context interrupted Multi-step Navigation

Avg 25 min investigation Core Requirements Unified View One-stop cluster-wide stat us overview Context Continuity Any object can drill down t o related systems Quick Action From issue discovery to fix in < 3 steps Target State Single Entry All platforms embedded Experiment ID Linked One-click drill-down Alert ▶ Diagnose ▶ Fix Full-process automation 图18-8 统一座舱的需求分析,从碎片化到一体化

18.5.2 五种集成策略

统一的路径不止一条。根据对业界主流平台和开源项目的调研,我们将统一可视化方案归纳为五种核心策略,每一种都有 其哲学出发点、技术架构和适用场景。

  1. 中央仪表盘策略 •哲学:将所有功能收拢到一个统一的门户中,提供跨工作流的深层集成。 •代表产品:Kubeflow Central Dashboard、Run:ai Platform、Apolo(NeuroPlatform) •核心特征:单一 Web 入口,所有功能作为 Portal 内的子模块;统一的命名空间/项目概念贯穿所有子系统;深度工作 流集成,例如从 Notebook → Pipeline → Experiment → Serving 在同一个 UI 中完成。 典型架构(以 Kubeflow Central Dashboard 为例)如图18-9所示。 Kubeflow Central Dashboa rd (React + Node.js) Notebooks Pipelines Experiments Models Serving Istio + OIDC K8s Namespace JupyterLab KFP UI Katib ML Metadata KServe Unified Auth Multi-tenant Isolation 图18-9 Kubeflow Central Dashboard 架构 所有子系统挂载在统一的命名空间路由下。 •优点:体验最统一、工作流串联最紧密 •缺点:需要所有子系统按统一框架开发,迁移成本极高;对已有系统不友好
  2. 实验中心策略 •哲学:AI 平台的核心业务对象是“训练实验”,所有可视化围绕实验展开。 •代表产品:Determined AI WebUI、Weights & Biases (W&B)、MLflow Tracking UI •核心特征:以 Experiment(实验)为一级导航对象;每个实验页面内聚合超参、指标曲线、日志、系统资源、模型产 出;跨实验的比较和筛选功能强大。 •适用场景:以模型训练为核心业务的团队,其他功能(数据、部署)相对次要 实验中心模式以实验 ID 为枢纽聚合所有相关信息,如图18-10所示。 Hyperparameters batch_size=128, lr=1e-4

Metrics loss, accuracy, throughput Logs Trial logs, profiler trace Experiment experiment_id=1822 System Metrics GPU util, memory, IB BW Checkpoints Model artifacts Metadata DB PostgreSQL 图18-10 Experiment-Centric 模式 3) 计算框架中心策略 •哲学:以统一的分布式计算框架为基础,可视化围绕计算资源管理展开。 •代表产品:KubeRay + Ray Dashboard、OpenSCOW(北大) •核心特征:Ray / KubeRay 作为统一计算抽象层,训练、数据处理、推理、调参都用 Ray;Ray Dashboard 提供统一 的 Job、Actor、Task 视图;所有工作负载的资源和日志都可以在同一个 UI 中定位。 •适用场景:已经从 PyTorch DDP 迁移到 Ray 框架的团队 •优点:天然统一,Ray 本身就是抽象层 •缺点:绑定 Ray 生态,需要全团队的技术栈迁移 4) GPU 资源中心策略 •哲学:GPU 是集群的“硬通货”,以物理 GPU 设备为核心组织视图。 •代表产品:HAMi WebUI(第四范式开源)、NVIDIA DCGM + 自研面板 •核心特征:物理拓扑视角,从机柜 → 服务器 → GPU Chip → 进程 的树状导航;GPU 分配热力图,哪些 GPU 被占用、 哪些空闲、哪些故障;支持 GPU 虚拟化(vGPU、MIG)的分层展示。 GPU 资源中心以物理拓扑组织 GPU 的分配和健康状态,如图18-11所示。 Datacenter Rack R01 (16 DGX) Rack R02 (16 DGX) DGX-01 (8×H800) DGX-02 (8×H800) GPU 0 GPU 1 GPU 2 GPU 3 GPU 4 GPU 5 GPU 6 GPU 7 Pod: train-1822 Pod: train-1822 Pod: train-1928 Pod: -- ERROR: XID 48 fault Pod: train-1822 Pod: train-1822 Pod: train-1822 Util: 98% Util: 97% Util: 45% Util: 0% idle Util: 98% Util: 98% Util: 96% 图18-11 GPU Resource-Centric 视图 5) 告警与自愈策略 •哲学:对于大规模集群,“稳定性”是压倒一切的首要目标。 •代表产品:Crusoe Command Center、Datadog AI Operations •核心特征:以告警(Alert)为一级对象,告警驱动一切操作;告警附带丰富的上下文(相关指标、日志、拓扑);自动 化 Runbook,告警触发 → 诊断 → 建议 → 一键执行修复;告警生命周期管理,触发 → 确认 → 处理中 → 已解决 → 复 盘。 •适用场景:GPU 规模 > 1,000 张、SLA 要求 > 99.5% 的生产集群 6) 五种策略对比 五种集成策略的横向对比如表18-41所示。 表18-41 五种集成策略对比 策略 核心哲 关键特征 理想场景 代表产品 成熟度 学 Central Dashboard 统一门 户 单一入口、命名空间贯穿、深度 从零搭建的新平台 Kubeflow、Run:ai、 ★★★ 工作流集成 Apolo ★★ Experiment-Centric 实验中心 实验 ID 聚合一切、强对比和追 以模型训练为核心 Determined AI、 踪能力 的团队 W&B、MLflow ★★★ ★☆ Compute 计算统 统一计算抽象、Ray 生态整合 已全面采用 Ray 的 KubeRay、OpenSCOW ★★★ Framework-Centric 一 团队 ☆☆ GPU Resource- 硬件视 物理拓扑导航、GPU 热力图、 GPU 运维和容量规 HAMi WebUI、DCGM ★★★ Centric 角 虚拟化分层 划 ☆☆ Alert & Self-Healing- 稳定性 告警驱动、自动 Runbook、生 大规模生产集群 Crusoe、Datadog ★★★ Centric 优先 命周期管理 ★☆ 说明:五种策略并非互斥。实际落地中,往往是 1~2 种策略主导 + 其他策略辅助 的混合模式。例如,Run:ai 本质上 是“Central Dashboard + GPU Resource-Centric”的融合。

18.5.3 统一可视化的架构模式

无论选择哪种集成策略,都需要面对一个核心的工程问题:如何在技术上把多个独立的 Web 前端“拼”在一起?以下是 五种主流的架构模式。

  1. 架构模式全景图 五种架构模式的全景如图18-12所示。 Five Architecture Patterns Pattern 5: BFF Backend-for-Frontend

(Go/Node.js) Det API Prometheus API Data Platform API Unified Frontend Pattern 4: Rancher Extension Rancher UI Extension Plugin (@rancher/shell) Proxy ▶ Det/Grafana Pattern 3: iFrame Embedding Main Frame Page (with sidebar/navigation)