万卡集群训练功耗剧烈波动,AI数据中心电力管理亟待突破
在AI大模型高速发展的推动下,训练规模已从单GPU走向数万GPU的集群级别。但与此同时,巨大的功耗波动正在成为制约算力中心可持续发展的隐忧。微软、OpenAI与英伟达近期联合发布的研究表明,单个训练作业的功率波动已高达数十兆瓦,对数据中心乃至电网安全都提出了前所未有的挑战。

以GPT-3(1750亿参数)、PaLM(5400亿参数)和Llama3.1(4050亿参数)等超大模型为例,其训练任务普遍依赖批次同步并行范式。这种模式下,功耗在两个阶段出现极端反差:
计算密集阶段:GPU执行前向和反向传播,满负荷运转,功耗逼近热设计功率极限;
通信密集阶段:进行梯度同步或模型检查点保存时,大量GPU处于空闲或低负载,功耗骤然回落。
这一高低功耗的交替循环,造成了“过山车式”的能耗曲线,不仅影响供电系统的稳定性,还可能增加运维和电力调度成本。
为应对这一挑战,研究提出需从多个维度展开探索:
在软件层面,通过任务调度和负载均衡优化功率曲线;
在硬件层面,改进GPU能效与动态电源管理能力;
在基础设施层面,提升数据中心供配电与储能系统的灵活性。
可以预见,随着AI训练规模的继续扩张,功耗管理将与算力优化同样重要。未来能否实现绿色、稳定的AI算力供给,电力管理的创新或将成为关键突破口。