KV 缓存:本地大模型上下文占多少显存

KV 缓存随上下文长度线性增长。标准 GQA 的算法:2 × 层数 × KV 头数 × 头维度 × 缓存 token 数 × 序列数 × 每元素字节数,FP16 每元素 2 字节。

KV 缓存:本地大模型上下文占多少显存

2 × 28 × 4 × 128 × 8192 × 1 × 2

KV Cache(原始数据)
0.4375 GiB

469,762,048 bytes

  • 注意力类型:gqa。支持程度:full。GQA 公式适用。
  • 查询头 28 个,KV 头 4 个。 如果按查询头算,会是 3.06 GiB。
  • 上下文类型:native。原生窗口 32,768。 没有记录扩展窗口。
  • 当前结果是估算,不是对应环境的实测。

结构

2
K + V
layers
层数
kvHeads
KV 头
headDim
头维度
cachedTokens
上下文长度(缓存 token 数)
sequences
同时运行的序列数
bytes
FP16/BF16 = 2 · FP8/INT8 = 1

估算占用

模型结构8,192 FP1632,768 FP16
Qwen2.5-7B-Instruct28 × 4 × 128469,762,048 (0.4375 GiB)1,879,048,192 (1.7500 GiB)
Qwen2.5-14B-Instruct48 × 8 × 1281,610,612,736 (1.5000 GiB)6,442,450,944 (6.00 GiB)
Qwen2.5-32B-Instruct64 × 8 × 1282,147,483,648 (2.00 GiB)8,589,934,592 (8.00 GiB)
Qwen3-8B36 × 8 × 1281,207,959,552 (1.1250 GiB)4,831,838,208 (4.5000 GiB)
计算器指南

下一步