Skip to content

清华大学 GPU 集群交付复盘:从架构到验收 ​

摘要:一次大型 GPU 集群交付,本质是把"业务需求"翻译成"架构决策",再把"架构决策"兑现成"可验收的系统"。本文复盘主导清华 GPU 集群交付的完整路径——需求、架构、验收、排障各环节的方法论,细节已脱敏,方法可复用。

1. 需求拆解:先问清楚"给谁用、跑什么" ​

交付最怕的是"客户说 100 卡,实际是 100 卡跑 3 种完全不同的负载"。需求阶段必须锁死四件事:

需求维度必须问清的问题影响决策
负载类型训练/推理/科学计算?各自占比?算力 vs 带宽 vs FP64 配比
并发规模同时跑几个作业?单作业最大卡数?调度策略、队列划分
数据规模数据集/检查点多大?读写模式?存储容量与带宽
用户水平研究人员/工程师/学生?软件栈复杂度、文档与培训投入

本例结论:以大规模 LLM 训练为主、多用户并发,科学计算为辅 → 算力优先、互联高配、存储按训练负载设计。

2. 架构决策:三大件 + 存储 + 调度 ​

2.1 算力与互联 ​

  • 节点规格:8 卡 GPU 节点(NVLink 全互联),TP=8 单节点打底——跨节点只跑 PP/DP,通信占比可控。
  • 节点间网络:IB 高速网络 + 胖树拓扑,满足 AllReduce 均匀散列;RoCE 方案需严格 PFC/ECN(见 NCCL 调优)。
  • 决策依据:用目标模型(如 70B)+ 并行策略倒推(见 分布式并行全景),而不是拍脑袋定卡数。

2.2 存储 ​

  • 并行文件系统(Lustre/GPFS 类)承载数据集与检查点,聚合带宽按"checkpoint 分钟级落盘"倒推(见 验收清单)。
  • 家目录/共享目录与高速存储分离,避免"小 IO 打爆大带宽"。

2.3 调度与软件栈 ​

  • Slurm 异构调度:GPU 节点 + CPU 节点混管,gres 配置 GPU 资源,队列按作业类型划分(训练队列/测试队列/交互队列)。
  • 容器运行时(Singularity/Apptainer 或 Enroot)+ 镜像仓库,保证环境可复现——可复现性是交付质量的底线。
  • 监控:GPU 利用率、温度、ECC、IB 流量统一接入,交付即移交监控,不留运维空白。

3. 四步验收:不达标不签字 ​

严格按照 验收清单 的流程执行:

Step 1  HPL 单节点算力        → 实测 ≥ 规格 95%
Step 2  Stream 显存带宽       → 实测 ≥ 规格 90%
Step 3  nccl-tests 跨节点互联 → 256 卡 busbw ≥ 40 GB/s(400G IB)
Step 4  72h 真实负载烤机      → 训练+推理+数据加载混合跑,无致命错误

本例关键数据:单节点 HPL 全部达标(无降频);跨节点 nccl-tests 初次 38 GB/s 略低于线,定位为 ECMP 散列不均,开启 enhanced hash 后 45 GB/s 达标。

4. 排障实录:三个代表性问题的处置 ​

4.1 ECMP 散列不均(互联层) ​

  • 现象:256 卡 nccl-tests 波动大,个别链路满载。
  • 定位:逐对测 2 节点正常 → 全集群测异常 → 交换机 telemetry 看流量分布。
  • 处置:开启 ECMP enhanced hash(l4 端口散列),重测恢复正常。教训:大规模基准必须在拓扑层验收,单点正常不代表集群正常。

4.2 数据加载 IO 瓶颈(存储层) ​

  • 现象:训练 GPU 利用率 60%,数据加载等待占比高。
  • 定位:nvidia-smi 看 GPU 空转 + 存储侧 IOPS 打满。
  • 处置:数据预取/缓存层 + 并行文件系统客户端数扩容,利用率回到 90%+。教训:算力验收通过 ≠ 训练跑得快,数据管线是隐藏瓶颈。

4.3 驱动与容器兼容(软件层) ​

  • 现象:部分用户容器起不来,报 CUDA 版本错误。
  • 定位:驱动支持版本 vs 镜像内 CUDA 运行时版本不匹配。
  • 处置:制定"驱动/CUDA/镜像兼容矩阵"文档 + 基础镜像统一维护。教训:软件栈标准化是交付后运维的核心资产(见 驱动生命周期)。

5. 交付清单(可复用) ​

  • [ ] 需求四问(负载/并发/数据/用户)已锁死
  • [ ] 算力/互联/存储按目标负载倒推,有据可查
  • [ ] 四步验收全部达标(HPL/Stream/nccl-tests/72h)
  • [ ] 拓扑层基准通过(256 卡跨机),非仅单点
  • [ ] 数据管线验证:训练 GPU 利用率 ≥ 85%
  • [ ] 软件栈标准化:镜像/驱动/CUDA 兼容矩阵文档化
  • [ ] 监控与告警移交,运维 runbook 就绪
  • [ ] 备份与检查点策略落地(恢复演练通过)

延伸阅读:北京大学 HPC 项目交付复盘(科学计算视角的补充)、NVIDIA 技术认证备考指南(能力验证路径)。

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