首页/AI/大模型部署与运维速查/KV 缓存是什么:显存的主要变量
本包目录(267 条)
部署形态与选型43 条推理环境与 GPU50 条
推理参数与优化50 条
显存与并发估算40 条服务化与网关34 条运维监控与排障50 条

KV 缓存是什么:显存的主要变量

KV 缓存保存每层注意力已生成 token 的 Key/Value 张量,避免重复计算:随序列长度、层数、KV 头数、精度与批大小线性增长,显存规划必须单独为它留空间。

量级直觉

量级直觉
KV 显存 ≈ 2 × 层数 × KV头数 × 头维度 × 序列长度 × 并发序列数 × 精度字节
# GQA 模型的 KV头数 小于注意力头数,按 KV头数计算
详细公式与算例见「显存与并发估算」分组;这里记住四个驱动变量:序列长度、并发数、精度、模型结构。

易错点

  • 只按权重估显存:长上下文与高并发下 KV 缓存可以超过权重占用,是线上 OOM 的常见来源

数据来源与参考

  • vLLM 官方文档:PagedAttention 与 KV 管理(docs.vllm.ai)
  • Hugging Face 文档:KV cache(huggingface.co/docs)
​‌‌​​‌​​​‌‌​‌‌‌‌​‌‌​​​‌‌​‌‌‌​​‌‌​​‌​‌‌‌​​‌‌‌​​​‌​‌‌​‌​​‌​‌‌‌​​​‌​‌‌​‌​​‌​‌‌​​​‌‌​‌‌‌​‌​​​‌‌​‌‌‌‌​​‌​‌‌‌​​‌‌​​​‌‌​‌‌​‌‌‌‌​‌‌​‌‌​‌