toc 目录

管理员 - 1

Object2
Object2 摆烂中
tagHome
arrow_back返回
Object2

AI的能源账单:训练一个大模型要烧掉多少电

大模型的训练和推理消耗大量能源,AI的快速发展引发了对环境影响的担忧。本文分析AI的能源消耗现状、优化方向以及行业如何平衡AI发展和可持续性。

AI的能源账单:训练一个大模型要烧掉多少电

AI的能源账单:训练一个大模型要烧掉多少电

训练一个大型语言模型消耗的电力,可能相当于一个小城市一个月的用电量。

这个数字不是夸张。GPT 级别的模型训练需要几千块 GPU 运行几个月。每块 GPU 的功耗在几百瓦,几千块 GPU 的总功耗就是兆瓦级。加上冷却系统的能耗,一个大型训练任务的总能耗是惊人的。

AI 的快速发展带来了巨大的能源需求,这个需求正在对全球的电力基础设施和碳排放目标产生影响。

AI 的能源消耗现状

Server infrastructure

AI 的能源消耗主要来自两个方面:训练和推理。

训练是一次性的成本。一个大模型从零开始训练一次,需要消耗大量的计算资源和电力。训练越大的模型、用越多的数据,消耗的能源越多。

推理是持续性的成本。模型训练完成后,每次用户请求都需要消耗计算资源。当模型被数百万用户使用时,推理的总能耗可能远超训练的能耗。

有估算显示,一次大模型推理消耗的电力大约是传统搜索引擎查询的十倍。当 AI 搜索逐渐替代传统搜索时,全球搜索的总能耗可能增加一个数量级。

数据中心是 AI 能耗的主要载体。全球数据中心的用电量已经占到全球总用电量的百分之几,而且在快速增长。AI 是这个增长的主要驱动力。

为什么 AI 这么耗能

AI 的高能耗根源于它的计算模式。

大模型的训练需要大量的矩阵运算。这些运算需要在 GPU 或者专用加速器上进行,每一步运算都需要读写大量的数据。数据在芯片内部和芯片之间的传输消耗了大量的能量。

模型越大,需要的计算量越大。模型参数从几十亿增长到几千亿,计算量不是线性增长,而是超线性增长。更大的模型需要更多的 GPU、更长的训练时间、更多的能源。

数据量也是能耗的来源。训练数据从几百 GB 增长到几 TB,数据的预处理、存储、加载都需要能源。

优化方向

面对 AI 的能源挑战,行业从多个方向在优化。

模型效率是最直接的方向。更高效的模型架构可以在保持性能的前提下减少计算量。混合专家模型(MoE)就是一个例子:它不是激活所有参数,而是根据输入只激活一部分,大幅降低了推理的计算量。

量化也是有效的优化。把模型参数从 FP16 降到 INT8 甚至 INT4,计算量和能耗都可以显著降低。

硬件效率也在持续提升。新一代的 GPU 和 AI 芯片在每瓦特性能上持续改善。专用的推理芯片比通用 GPU 在推理场景下能效更高。

数据中心的能效优化也很重要。更好的冷却技术(液冷、自然冷却)、更高的服务器利用率、可再生能源的使用,都可以降低数据中心的碳足迹。

可再生能源的角色

Data center architecture

使用可再生能源是降低 AI 碳足迹的最直接方式。

各大云厂商都在承诺使用 100% 可再生能源。Google、微软、亚马逊都签署了大量可再生能源购买协议。

但可再生能源有间歇性问题。太阳能只在白天有,风能取决于天气。数据中心需要 7x24 不间断供电,完全依赖可再生能源在技术上有挑战。

储能技术和智能电网是解决方案之一。在可再生能源充裕时充电,在不足时放电。AI 训练任务也可以调度到可再生能源充裕的时段执行。

核能也在重新获得关注。核能提供稳定、低碳的电力,适合作为数据中心的基荷电源。微软已经签署了核电购买协议,Google 也在投资小型模块化核反应堆。

行业的责任

AI 行业需要正视自己的能源责任。

透明度是第一步。公开 AI 模型训练和推理的能耗数据,让用户和投资者了解 AI 的环境成本。目前这方面的透明度还很低。

能效应该是模型评估的重要指标。不只看模型的准确率,还要看每单位准确率消耗的能源。一个准确率高 1% 但能耗高 10 倍的模型,不一定比能效更高的模型更优。

碳补偿也是可行的措施。对于无法避免的碳排放,通过植树、碳捕获等方式进行补偿。

Cloud server technology

我的判断

AI 的能源挑战是真实的,但不是不可解决的。技术优化、可再生能源、核能等多种方案可以共同缓解这个问题。

关键是要在 AI 发展和可持续性之间找到平衡。不能因为能源问题就停止 AI 的发展,也不能因为 AI 的价值就忽视能源问题。

对于 AI 从业者来说,关注能效是一种责任。在模型选择、架构设计、部署策略中考虑能效因素,不只是为了省钱,也是为了环境。

对于政策制定者来说,需要在鼓励 AI 创新和保障能源安全之间找到平衡。碳税、能效标准、可再生能源激励等政策工具可以引导行业向更可持续的方向发展。

AI 的未来应该是既强大又绿色的。这两个目标不是矛盾的,技术进步可以同时实现两者。

吉祥物