量级直觉
量级直觉
KV 显存 ≈ 2 × 层数 × KV头数 × 头维度 × 序列长度 × 并发序列数 × 精度字节
# GQA 模型的 KV头数 小于注意力头数,按 KV头数计算详细公式与算例见「显存与并发估算」分组;这里记住四个驱动变量:序列长度、并发数、精度、模型结构。
KV 缓存保存每层注意力已生成 token 的 Key/Value 张量,避免重复计算:随序列长度、层数、KV 头数、精度与批大小线性增长,显存规划必须单独为它留空间。
KV 显存 ≈ 2 × 层数 × KV头数 × 头维度 × 序列长度 × 并发序列数 × 精度字节
# GQA 模型的 KV头数 小于注意力头数,按 KV头数计算