2、基础知识:AI算力集群的组成、能耗模型与能效指标

2.1 AI算力集群的物理组成与能耗边界

要理解能效调优,首先必须明确“算力集群”的物理边界。一个典型的AI训练/推理集群并非只有GPU服务器,而是由IT设备与基础设施设备共同构成的复杂系统。从能耗审计的角度,我们通常将集群划分为以下层次:

  • 计算层(IT负载):GPU/AI加速卡、CPU、内存、本地NVMe SSD。这是真正执行矩阵乘法和张量运算的部分,功耗随负载剧烈波动。
  • 网络层:Leaf/Spine交换机、光模块、RDMA网卡(如CX-6/CX-7)。在万卡集群中,网络功耗可占IT总功耗的10%~15%。
  • 存储与管理层:并行文件系统存储节点、登录节点、管理网交换机。
  • 供配电系统:UPS、PDU、变压器、母线槽。其效率直接影响电能从电网到芯片的转化率。
  • 制冷散热系统:CRAH/CRAH、冷水机组、冷却塔、液冷CDU(冷量分配单元)、二次侧管路。
能耗边界陷阱:许多团队在统计PUE时只测量IT机房内的UPS输出端,而忽略了制冷机房内的冷水机组、冷却塔功耗。这会导致PUE虚低,掩盖真实能效问题。正确的边界应包含所有支撑IT设备运行的辅助能源消耗。

下图展示了AI算力集群的典型能耗拓扑与测量点位置,这是后续计算PUE/DCiE的基础:

AI算力集群能耗拓扑与PUE测量边界 电网输入 10kV/380V AC 总输入电能 (P_total) 电表M1(边界点) 供配电系统 变压器→UPS→PDU→母线 损耗:约5%~8% 制冷散热系统 冷水机组+CDU+冷却塔 损耗:占PUE 0.1~0.3 辅助照明/监控 照明、动环监控、门禁 损耗:约1%~2% IT设备(计算/网络/存储) GPU服务器 8×H100/A100 功耗:6.5~10kW 占IT功耗 70%~80% 网络设备 Spine/Leaf交换机 400G光模块 占IT功耗 10%~15% 存储与管理 并行文件系统 登录/管理节点 占IT功耗 5%~10% IT设备功耗 (P_IT) —— 电表M2 通常为UPS输出端或PDU输入端 热量排出 PUE = P_total / P_IT = (P_IT + P_制冷 + P_供配电损耗 + P_辅助) / P_IT DCiE = 1 / PUE = P_IT / P_total × 100%

2.2 能耗模型:从芯片到集群的功率分解

AI集群的能耗模型遵循“功率叠加”与“时间积分”两条主线。瞬时功率决定配电容量,累计电量决定电费成本。在能效调优中,我们更关注功率的时间分布特性——AI训练任务具有明显的波峰-波谷特征:

  • 数据加载阶段:GPU空闲,CPU/存储IO繁忙,整机功耗约为峰值的30%~40%。
  • 前向计算阶段:GPU利用率攀升至80%以上,功耗快速上升。
  • 反向传播与梯度同步:GPU计算与网络通信叠加,达到功耗峰值(如H100满载约700W/卡)。
  • 检查点保存(Checkpoint):GPU暂停计算,存储写入爆发,功耗短暂回落。

一个实用的集群级能耗估算公式如下:

P_cluster = N_gpu × P_gpu_avg + N_cpu × P_cpu_avg + P_network + P_storage + P_loss
其中:
  P_gpu_avg = P_idle + (P_max - P_idle) × GPU_utilization
  P_loss = P_IT × (1/η_power - 1) + P_cooling
  η_power 为供配电效率(通常0.92~0.95)
  P_cooling 由制冷系统COP(能效比)决定:P_cooling = Q_heat / COP
案例:某集群有512张H100 GPU,单卡峰值功耗700W,空闲功耗100W。实测训练时平均GPU利用率为85%。则GPU层功耗为:512 × (100 + (700-100)×0.85) = 512 × 610W ≈ 312kW。加上CPU/内存/网络约60kW,IT总功耗约372kW。若PUE=1.25,则总输入功耗为465kW,制冷与配电损耗为93kW。

2.3 核心能效指标:PUE、DCiE与部分PUE

能效指标是衡量集群“有多少电真正用于计算”的标尺。以下是行业最核心的几项指标:

指标 定义/公式 理想范围 适用场景
PUE (Power Usage Effectiveness) PUE = 总输入电能 / IT设备电能 1.1~1.3(风冷)
1.05~1.15(液冷)
衡量基础设施(制冷+供配电)开销
DCiE (Data Center Infrastructure Efficiency) DCiE = 1 / PUE × 100% 75%~90% PUE的倒数形式,直观表示IT占比
pPUE (partial PUE) pPUE = 子系统输入电能 / IT设备电能 制冷pPUE ≤ 0.15
供配电pPUE ≤ 0.08
定位具体子系统(制冷/配电)的效率短板
ITEU (IT Equipment Utilization) ITEU = 实际算力功耗 / IT设备额定功耗 ≥ 0.6(训练负载) 衡量IT设备本身是否被充分利用
GPU利用率 GPU SM活跃时间 / 总时间 ≥ 85%(训练) 反映计算任务对芯片的利用程度
重要误区:PUE低并不等于集群能效高。一个PUE=1.1但GPU利用率仅30%的集群,其“有效算力功耗”远低于PUE=1.3但GPU利用率90%的集群。因此,业界逐渐引入“算力能效”综合指标(如每瓦特每秒浮点运算次数,即FLOPS/W),将PUE与IT设备利用率联合评估。

2.4 能效指标测量方法与常见误区

正确的能效测量需要遵循以下流程,否则数据将失去可比性:

  1. 确定测量边界:明确电表M1(总输入)和M2(IT输入)的物理位置,绘制配电单线图。
  2. 同步时间窗口:PUE是时间平均值,建议以15分钟为粒度采集,至少持续24小时覆盖完整训练周期。
  3. 区分负载类型:训练任务(高功耗持续型)与推理任务(低功耗间歇型)的PUE差异显著,需分开统计。
  4. 记录环境参数:室外温度、冷却水温度、IT负载率需同步记录,否则无法解释PUE波动原因。
  5. 计算与归档:按公式计算PUE/DCiE,并生成趋势曲线用于调优前后对比。

以下是测量与解读中的典型误区:

  • 误区一:只看瞬时PUE。夜间低负载时PUE可能高达2.0,而白天满载时仅1.15,取瞬时值会严重误导决策。
  • 误区二:忽略IT负载率对PUE的影响。PUE随负载率升高而降低(因为基础设施功耗相对固定),因此对比PUE必须注明负载率。
  • 误区三:将芯片功耗与服务器功耗混淆。GPU功耗≠服务器功耗,后者还包括CPU、内存、电源转换损耗(约8%~12%)。
  • 误区四:未考虑液冷系统的二次能耗。液冷虽然降低了PUE,但CDU泵、二次侧管路泵、冷却塔风机同样耗电,需计入总输入。
实操建议:在集群部署时,为每个GPU服务器机柜安装智能PDU(支持每端口计量),并配置DCIM(数据中心基础设施管理)平台。这样可实时获取P_IT与P_total,自动计算PUE并生成告警。对于液冷集群,还需在CDU入口安装热量计,以验证实际带走的热量是否与IT功耗匹配(热量平衡校验)。