Codestral 22B — Besoins VRAM, matrice de quantification et guide GPU

Codestral 22B (Mistral, Mistral Non-Production) repose sur une architecture Transformer dense comptant 22.2B paramètres résidents (22.2B actifs par token), 56 couches et Attention à requêtes groupées (GQA) prenant en charge jusqu’à 32,768 tokens de contexte.

Simulateur interactif de mémoire et de contexte pour Codestral 22B

32,768 tokens

Répartition mémoire et plafond de bande passante

Ajusté (>90 % utile)
Poids du modèle: 59.2% KV: 33.1% Overhead: 7.7%
Mémoire totale requise
21.18 Go
Allocation par périphérique: 21.18 Go
Poids du modèle
12.53 Go
Cache KV (FP16): 7.00 Go
Décodage estimé (1 flux)
27.9 tok/s
Surcharge runtime et TP: 1.64 Go
Contexte max avant OOM
43,438
tokens

La compression KV architecturale économise 83.3 % par rapport au MHA FP16 naïf (42.00 Go).

Matrice VRAM par précision et quantification (contexte 8K / 32K / 128K)

Mémoire GPU totale requise selon la précision des poids et la longueur du contexte (batch = 1, cache KV FP16)
Précision / Format bpw effectifs Poids seuls Total @ ctx 8K Total @ ctx 32K Total @ ctx 128K Palier recommandé (@ 32K)
FP16 / BF16 16.00 bpw 41.35 Go 45.61 Go 50.86 Go 71.86 Go NVIDIA A100 80GB
FP8 8.50 bpw 21.97 Go 25.64 Go 30.89 Go 51.89 Go RTX 5090 32GB
Q8_0 8.50 bpw 21.97 Go 25.64 Go 30.89 Go 51.89 Go RTX 5090 32GB
Q6_K 6.56 bpw 16.95 Go 20.48 Go 25.73 Go 46.73 Go RTX 5090 32GB
Q5_K_M 5.69 bpw 14.71 Go 18.16 Go 23.41 Go 44.41 Go RTX 5090 32GB
Q4_K_M 4.85 bpw 12.53 Go 15.93 Go 21.18 Go 42.18 Go RTX 3090 24GB
AWQ INT4 4.50 bpw 11.63 Go 14.99 Go 20.24 Go 41.24 Go RTX 3090 24GB
IQ4_XS 4.25 bpw 10.98 Go 14.33 Go 19.58 Go 40.58 Go RTX 3090 24GB

Compatibilité matérielle et vitesse de décodage (Codestral 22B @ Q4_K_M, contexte 32K)

Statut de compatibilité, utilisation mémoire et débit de décodage estimé sur les GPU phares et Apple Silicon
Palier matériel Mémoire et bande passante Requis (32K) Compatibilité Débit décodage est.
RTX 3090 24GB 24 Go · 936 GB/s 21.18 Go (88%) Ajusté (>90 % utile) 27.9 tok/s
RTX 4090 24GB 24 Go · 1008 GB/s 21.18 Go (88%) Ajusté (>90 % utile) 30.1 tok/s
RTX 5090 32GB 32 Go · 1792 GB/s 21.18 Go (66%) Tient confortablement 53.4 tok/s
Apple M-Max 64GB 64 Go · 546 GB/s 20.65 Go (32%) Tient confortablement 15.3 tok/s
Mac Studio 128GB 128 Go · 546 GB/s 20.65 Go (16%) Tient confortablement 15.3 tok/s
Mac Studio 192GB 192 Go · 800 GB/s 20.65 Go (11%) Tient confortablement 22.5 tok/s
NVIDIA A100 80GB 80 Go · 2039 GB/s 21.18 Go (27%) Tient confortablement 64.4 tok/s
NVIDIA H100 80GB 80 Go · 3350 GB/s 21.18 Go (27%) Tient confortablement 105.8 tok/s

Paramètres vérifiés de l’architecture Transformer

Checkpoint ID
mistralai/Codestral-22B-v0.1
Ollama Tag
codestral:22b
License / Release
Mistral Non-Production · 2024-Q3
Topologie et attention
Transformer dense
Mécanisme d’attention
Attention à requêtes groupées (GQA)
Paramètres totaux (B)
22.2B
Paramètres actifs / token (B)
22.2B
Couches cachées / Dimension cachée
56 / 6144
Têtes de requête / Têtes KV
48Q / 8KV (d=128)
Vocab / Max Position
32,768 / 32,768

Les formats quantifiés stockent des facteurs d’échelle et des points zéro par bloc aux côtés des poids entiers compressés. Un checkpoint GGUF Q4_K_M 4 bits occupe en moyenne 4.85 bits effectifs par paramètre (et non 4.00), tandis que Q8_0 atteint 8.50 bpw. Pour les architectures Mixture-of-Experts, l’ensemble des experts routés et partagés reste résident en mémoire afin d’éviter la pagination hôte à chaque token.

L’attention à requêtes groupées standard alloue 2 × L × H_kv × d_head × S × B × bytes_kv sur l’ensemble des couches. L’attention latente multi-têtes de DeepSeek (MLA) compresse clés et valeurs en L × (d_kv_lora + d_qk_rope) × S × B × bytes_kv, réduisant de plus de 90 % l’empreinte mémoire sur long contexte. Les modèles hybrides à fenêtre glissante plafonnent les couches locales à min(S, window_size) tout en faisant croître linéairement les couches globales avec S.

Commandes rapides vLLM et Ollama pour Codestral 22B

CLI vLLM (RTX 3090 24GB)
vllm serve mistralai/Codestral-22B-v0.1 \
  --tensor-parallel-size 1 \
  --max-model-len 32768 \
  --max-num-seqs 8 \
  --gpu-memory-utilization 0.95 \
  --kv-cache-dtype fp8 \
  --enable-prefix-caching \
  --port 8000
Modelfile Ollama
# Save as Modelfile, then run: ollama create codestral-22b-custom -f Modelfile
FROM codestral:22b
PARAMETER num_ctx 32768
PARAMETER num_batch 512
PARAMETER flash_attn true

# Environment variables for ollama serve:
# OLLAMA_NUM_PARALLEL=8 OLLAMA_FLASH_ATTENTION=1 OLLAMA_KV_CACHE_TYPE=q8_0 ollama serve

Comparer les modèles proches et de taille similaire à poids ouverts