Llama 3.1 405B Instruct — Besoins VRAM, matrice de quantification et guide GPU

Llama 3.1 405B Instruct (Meta Llama, Llama 3.1 Community) repose sur une architecture Transformer dense comptant 405B paramètres résidents (405B actifs par token), 126 couches et Attention à requêtes groupées (GQA) prenant en charge jusqu’à 131,072 tokens de contexte.

Simulateur interactif de mémoire et de contexte pour Llama 3.1 405B Instruct

32,768 tokens

Répartition mémoire et plafond de bande passante

Tient confortablement
Poids du modèle: 91.7% KV: 6.3% Overhead: 2%
Mémoire totale requise
249.40 Go
Allocation par périphérique: 249.40 Go
Poids du modèle
228.67 Go
Cache KV (FP16): 15.75 Go
Décodage estimé (1 flux)
1.8 tok/s
Surcharge runtime et TP: 4.98 Go
Contexte max avant OOM
446,229
tokens

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

macOS plafonne par défaut la mémoire GPU verrouillée vers 384.0 Go (75 %) ; augmenter iogpu.wired_limit_mb débloque jusqu’à 450.6 Go (88 %).

sudo sysctl iogpu.wired_limit_mb=461373

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 754.37 Go 771.17 Go 782.99 Go 830.24 Go 8× NVIDIA H100 80GB
FP8 8.50 bpw 400.76 Go 412.26 Go 424.07 Go 471.32 Go 8× NVIDIA H100 80GB
Q8_0 8.50 bpw 400.76 Go 412.26 Go 424.07 Go 471.32 Go 8× NVIDIA H100 80GB
Q6_K 6.56 bpw 309.29 Go 319.42 Go 331.23 Go 378.48 Go Mac Studio 512GB
Q5_K_M 5.69 bpw 268.27 Go 277.78 Go 289.60 Go 336.85 Go Mac Studio 512GB
Q4_K_M 4.85 bpw 228.67 Go 237.59 Go 249.40 Go 296.65 Go Mac Studio 512GB
AWQ INT4 4.50 bpw 212.17 Go 220.84 Go 232.65 Go 279.90 Go Mac Studio 512GB
IQ4_XS 4.25 bpw 200.38 Go 208.87 Go 220.69 Go 267.94 Go Mac Studio 512GB

Compatibilité matérielle et vitesse de décodage (Llama 3.1 405B Instruct @ 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 253.17 Go (1055%) Mémoire insuffisante (OOM) —
RTX 4090 24GB 24 Go · 1008 GB/s 253.17 Go (1055%) Mémoire insuffisante (OOM) —
RTX 5090 32GB 32 Go · 1792 GB/s 253.17 Go (791%) Mémoire insuffisante (OOM) —
Apple M-Max 64GB 64 Go · 546 GB/s 249.40 Go (390%) Mémoire insuffisante (OOM) —
Mac Studio 128GB 128 Go · 546 GB/s 249.40 Go (195%) Mémoire insuffisante (OOM) —
Mac Studio 192GB 192 Go · 800 GB/s 249.40 Go (130%) Mémoire insuffisante (OOM) —
NVIDIA A100 80GB 80 Go · 2039 GB/s 253.17 Go (317%) Mémoire insuffisante (OOM) —
NVIDIA H100 80GB 80 Go · 3350 GB/s 253.17 Go (317%) Mémoire insuffisante (OOM) —

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

Checkpoint ID
meta-llama/Llama-3.1-405B-Instruct
Ollama Tag
llama3.1:405b
License / Release
Llama 3.1 Community · 2024-Q3
Topologie et attention
Transformer dense
Mécanisme d’attention
Attention à requêtes groupées (GQA)
Paramètres totaux (B)
405B
Paramètres actifs / token (B)
405B
Couches cachées / Dimension cachée
126 / 16384
Têtes de requête / Têtes KV
128Q / 8KV (d=128)
Vocab / Max Position
128,256 / 131,072

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 Llama 3.1 405B Instruct

CLI vLLM (Mac Studio 512GB)
vllm serve meta-llama/Llama-3.1-405B-Instruct \
  --tensor-parallel-size 1 \
  --max-model-len 32768 \
  --max-num-seqs 8 \
  --gpu-memory-utilization 0.80 \
  --kv-cache-dtype fp8 \
  --enable-prefix-caching \
  --port 8000
Modelfile Ollama
# Optional macOS unified-memory ceiling override before launching:
# sudo sysctl iogpu.wired_limit_mb=461373

# Save as Modelfile, then run: ollama create llama-3-1-405b-custom -f Modelfile
FROM llama3.1:405b
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