跳转至

尾声 训练、推理与部署综合设计

第二册的验收不是"会用某个框架",而是能独立完成下面三道题,并让每个选择都落到数字上。

题一:预训练集群

给定 70B 稠密模型、8K 序列、1024 张 GPU、固定 token 预算:

  1. 用第 1 章的方法算出单卡放不下的是哪一项。
  2. 按第 7.1 节的决策顺序定 TP / CP / FSDP / DP,并说明为什么 PP 用或不用。
  3. 选优化器(第 8 章),说明它对显存和收敛的影响,以及学习率要怎么定。
  4. 估算每卡显存、每步通信量、step 时间和 MFU。
  5. 设计数据管线、检查点周期(用 goodput 求最优)、故障恢复和监控项。

题二:在线推理服务

给定一个 MoE 模型、真实的输入输出长度分布、TTFT 和 TPOT 的 SLO:

  1. 决定副本数、TP、EP,以及要不要 PD 分离(并说明判据)。
  2. 算出权重和 KV 的显存占用,推出最大并发。
  3. 设计调度:token 预算、分块 Prefill、前缀缓存策略、接纳控制和降级链。
  4. 判断投机解码在哪个并发区间开、哪个区间关。
  5. 用 goodput 和每百万 token 成本验收,给出延迟—吞吐曲线和容量建议。

题三:端侧交付

给定目标 SoC、可用内存、功耗预算和质量下限:

  1. 选量化精度和要保高精度的层,说明依据。
  2. 确认 runtime 的算子覆盖,排查可能的回退点。
  3. 打包成单一制品(图、权重、tokenizer、采样、版本)。
  4. 在真机上完成长时间性能、峰值内存、温控和质量验收。

你现在应该能回答

  • 这个方案为什么放得下?每一项显存花在哪?
  • 它为什么是这个速度?瓶颈是算力、带宽、容量还是排队?
  • 流量翻倍,最先坏的是哪里?
  • 某项优化省了什么、把代价转移到了哪里、在什么工作点上会反转?
  • 出现 OOM、吞吐下降、p99 恶化或质量回退时,按什么顺序排查?

能用数字回答这五个问题,而不是报框架名,才算真正掌握了工业级大模型技术。

下一册:模型能训能跑之后,怎么把它改成你的——后训练、LoRA、评测、harness 与 Agent。