腾讯面试题更新 2026-08-05

用 DeepSpeed ZeRO-3 微调 Qwen2-72B,每张卡占用显存大概是多少?请估算并解释为什么占这么多显存。

腾讯人工智能互联网/IT问题拆解技术原理方案权衡DeepSpeed

考察说明

考察对 ZeRO-3 显存优化原理的理解和实际估算能力

回答思路

  1. 能根据 ZeRO-3 将模型参数、梯度和优化器状态切分到各卡进行计算
  2. 能估算基础显存:模型权重(约 144GB)、梯度和优化器状态(如 Adam 的额外开销)
  3. 能考虑激活值、微调时的额外显存开销,以及 ZeRO-3 的 offload 选项
  4. 能给出合理的每卡显存范围并说明计算依据