Mistral Nemo 12B Instruct: how much VRAM does it need?

Mistral Nemo 12B Instruct has 12.25B parameters in 40 layers. It uses grouped-query attention: 32 query heads share 8 key-value heads, so its KV cache is 4× smaller than full multi-head attention would need. It takes up to 131,072 tokens of context; one user at that length needs 20.00 GiB of KV cache at FP16 on top of the weights.

BF16 · 8k context · one user
26.23 GiB

22.82 GiB of weights, 1.25 GiB of KV cache and 2.17 GiB of vLLM's overhead. At INT4 the same workload needs 11.31 GiB.

  • 40 layers
  • hidden 5120
  • 32/8 heads
  • head_dim 128
  • 128k max context

Mistral Nemo 12B Instruct memory by precision and context

Precision8k context32k context128k context
BF1626.23 GiBKV cache 1.25 GiB29.98 GiBKV cache 5.00 GiB44.98 GiBKV cache 20.00 GiB
FP816.07 GiBKV cache 1.25 GiB19.82 GiBKV cache 5.00 GiB34.82 GiBKV cache 20.00 GiB
INT411.31 GiBKV cache 1.25 GiB15.06 GiBKV cache 5.00 GiB30.06 GiBKV cache 20.00 GiB

One user, FP16 KV cache and vLLM's overhead on one GPU at its defaults for a card under 70 GB. Quantizing shrinks the weights, not the KV cache, so the saving narrows as the context grows.

Which single GPUs hold it

At 8k context for one user, the smallest cards first. A discrete card is budgeted at vLLM's default 92% of its memory, a unified-memory device at the share its GPU can address, and each carries vLLM's overhead at its own defaults.

How many users fit

GPUBF16FP8INT4
RTX 4090does not fit59
RTX 509031115
RTX 6000 Ada142326
H100 SXM 80GB374649
RTX PRO 6000 Blackwell495761
AMD Ryzen AI Max+ 395 (128GB)505862
H200 141GB829094
B200 180GB110118122

Users at 8k context each, on one card. ✗: the weights leave no room for even one.

Fine-tuning Mistral Nemo 12B Instruct

MethodMemorySmallest GPU
Full fine-tune184.3 GiBApple M3 Ultra (256GB)
LoRA (rank 16, attention)24.87 GiBRTX 5090
QLoRA (rank 16, attention)7.94 GiBRTX 3060

Sequence length 2,048, batch size 1, gradient checkpointing on, Adam. Full fine-tuning keeps 16 bytes per parameter before activations; QLoRA stores the frozen base in 4-bit NF4.

Where the numbers come from

Weights are parameters × bytes per parameter. The KV cache is 2 × layers × key-value heads × head dimension × bytes × context × users. For Mistral Nemo 12B Instruct that is 160.0 KiB per token at FP16. These are capacity-planning figures: the serving engine — vLLM, TGI, llama.cpp — and its paged-attention efficiency still move the real number.

More on KV cache size and on GQA, MQA and MLA.

Work with Mistral Nemo 12B Instruct

Other Mistral models