互联网/IT行业面试题更新 2026-08-05

DeepSpeed的ZeRO优化器在不同阶段(Stage 1、2、3)分别能节省哪些显存?各阶段的主要区别是什么?

网易人工智能互联网/IT技术原理技术选型方案权衡DeepSpeedPyTorch

考察说明

考察对分布式训练显存优化原理的理解

回答思路

  1. 能说明各阶段省去的数据类型和通信代价
  2. 能解释Stage 1省去优化器状态的条件
  3. 能解释Stage 2进一步省去梯度的条件
  4. 能说明Stage 3以及CPU offload省去参数及更高通信开销
本题已收录答题指导

本题附完整参考答案与评分标准

登录后可查看结构化答题指导;也可以直接开一场模拟面试,AI 面试官用本题实时追问并给出评分。