首页/AI/大模型部署与运维速查/每 token KV 心算算例
本包目录(267 条)
部署形态与选型43 条推理环境与 GPU50 条推理参数与优化50 条
显存与并发估算40 条
服务化与网关34 条运维监控与排障50 条

每 token KV 心算算例

把公式压成「每 token 大小」做心算:拿到层数、KV 头数与头维度后一次算出量级,再乘窗口与并发;fp8 KV 直接减半。

算例:80 层 / 8 KV 头 / 128 维 / FP16

算例:80 层 / 8 KV 头 / 128 维 / FP16
每 token:2 × 80 × 8 × 128 × 2 B ≈ 320 KiB
8K 窗口:320 KiB × 8192 ≈ 2.5 GiB / 序列
16 并发:≈ 40 GiB
(KV 用 fp8 存储时字节数减半)
同样算法换成 4K 窗口或 32 并发即可快速得到另一张表:先心算再上机验证。

易错点

  • 用注意力头数代替 KV 头数:GQA 模型会把 KV 高估数倍,容量规划随之失真

数据来源与参考

  • Hugging Face 模型配置文档(huggingface.co/docs)
  • vLLM 官方文档:FP8 KV cache(docs.vllm.ai)
​‌‌​​‌​​​‌‌​‌‌‌‌​‌‌​​​‌‌​‌‌‌​​‌‌​​‌​‌‌‌​​‌‌‌​​​‌​‌‌​‌​​‌​‌‌‌​​​‌​‌‌​‌​​‌​‌‌​​​‌‌​‌‌‌​‌​​​‌‌​‌‌‌‌​​‌​‌‌‌​​‌‌​​​‌‌​‌‌​‌‌‌‌​‌‌​‌‌​‌