LLM 系统分析方法论(五):训练显存估算
训练显存完整估算:从单卡四笔账(权重/优化器/梯度/激活)出发,叠加 TP/PP/DP/CP/EP 并行折扣,结合 ZeRO/FSDP、Gradient Checkpointing、Offload 建立训练显存体系。含 M3 完整案例和多模态/LoRA 微调场景。
训练显存完整估算:从单卡四笔账(权重/优化器/梯度/激活)出发,叠加 TP/PP/DP/CP/EP 并行折扣,结合 ZeRO/FSDP、Gradient Checkpointing、Offload 建立训练显存体系。含 M3 完整案例和多模态/LoRA 微调场景。
训练通信完整分析:从物理原理到框架实现,覆盖 TP/PP/DP/EP/CP/FSDP2 六种并行维度的通信模式、时间线建模和掩盖策略。含 M3 完整 step time 推演和 Dense 70B/M3 MoE 多场景实战。跨 NVIDIA + Ascend 双平台。
从 config.json 到参数量、FLOPs、KV Cache、推理显存的完整计算推导。覆盖 Full Attention / MSA / MLA / Mamba-2 / SWA / GDN 六种注意力架构。第一篇:矩阵乘法基础与参数分解。
KV Cache 原理与公式推导,覆盖 GQA / MLA / MSA / Mamba-2 四种架构的缓存策略;推理显存完整拆解,包括权重、KV Cache、激活值的显存占用计算。
FLOPs 完整估算:从矩阵乘法到 Attention 到 FFN,覆盖 Full Attention / MSA / MLA / Mamba-2 / GDN 六种注意力架构。