gpt-oss 120B kaç GB VRAM ister?

gpt-oss 120B, 36 katmanda 116,83B parametreye sahiptir. Grouped-query attention kullanır: 64 query head, 8 key-value head'i paylaşır; bu yüzden KV cache'i tam multi-head attention'a göre 8× küçüktür. Mixture-of-Experts modelidir: toplam 116,83B parametrenin token başına yaklaşık 5,10B kadarı aktiftir; bellek hepsini tutmak zorundadır, her token ise yalnızca aktif kısımdan geçer. En fazla 131.072 token context alır; bu uzunlukta tek bir kullanıcı, ağırlıkların üstüne FP16'da 4,65 GiB KV cache ister.

BF16 · 8k context · tek kullanıcı
220,1 GiB

217,6 GiB ağırlık, 437 MiB KV cache ve 2,04 GiB vLLM'in ek yükü. Aynı iş yükü INT4'te 63,24 GiB ister.

  • 36 katman
  • hidden 2880
  • 64/8 heads
  • head_dim 64
  • 128k maks context
  • MoE

Hassasiyet ve context'e göre gpt-oss 120B belleği

Hassasiyet8k context32k context128k context
BF16220,1 GiBKV cache 437 MiB220,9 GiBKV cache 1,27 GiB224,3 GiBKV cache 4,65 GiB
FP8112,4 GiBKV cache 437 MiB113,2 GiBKV cache 1,27 GiB116,6 GiBKV cache 4,65 GiB
INT463,24 GiBKV cache 437 MiB64,08 GiBKV cache 1,27 GiB67,46 GiBKV cache 4,65 GiB

Tek kullanıcı, FP16 KV cache ve 70 GB altı bir kartın varsayılanlarıyla tek GPU'da vLLM'in ek yükü. Quantization ağırlıkları küçültür, KV cache'i değil; bu yüzden context büyüdükçe kazanç daralır.

Onu tek başına taşıyan GPU'lar

8k context ve tek kullanıcı için, en küçük kartlar önce. Ayrık kartlarda vLLM'in varsayılanı olan bellek payı %92, unified bellekli cihazlarda GPU'nun kullanabildiği pay hesaba katılır; her cihaz kendi varsayılanlarıyla vLLM'in ek yükünü taşır.

Kaç kullanıcı sığar

GPUBF16FP8INT4
RTX 4090sığmazsığmazsığmaz
RTX 5090sığmazsığmazsığmaz
RTX 6000 Adasığmazsığmazsığmaz
H100 SXM 80GBsığmazsığmaz15
RTX PRO 6000 Blackwellsığmazsığmaz32
AMD Ryzen AI Max+ 395 (128GB)sığmazsığmaz29
H200 141GBsığmaz27114
B200 180GBsığmaz88166

Kullanıcı başına 8k context ile, tek kartta. ✗: ağırlıklar tek bir kullanıcıya bile yer bırakmıyor.

gpt-oss 120B ince ayarı

YöntemBellekEn küçük GPU
Tam ince ayar1742,3 GiBbirden çok GPU, paylaştırılmış
LoRA (rank 16, attention)219,1 GiBAMD MI325X
QLoRA (rank 16, attention)57,68 GiBAMD MI210

Dizi uzunluğu 2.048, batch 1, gradient checkpointing açık, Adam. Tam ince ayar aktivasyonlardan önce parametre başına 16 byte tutar; QLoRA dondurulmuş temel modeli 4-bit NF4 olarak saklar.

Sayılar nereden geliyor

Ağırlıklar parametre × parametre başına byte olur. KV cache 2 × katman × key-value head × head boyutu × byte × context × kullanıcı olur. gpt-oss 120B için bu, FP16'da token başına 72,0 KiB eder. 36 katmanının 18 tanesi son 128 token'lık kayan pencereye bakar ve bundan fazlasını tutmaz; pencereden sonra eklenen her token, diğer 18 katmanın payı olan 36,0 KiB ekler. Bunlar kapasite planlama sayılarıdır: servis motoru — vLLM, TGI, llama.cpp — ve paged-attention verimliliği gerçek değeri yine değiştirir.

Daha fazlası: KV cache boyutu ve GQA, MQA ve MLA.

gpt-oss 120B ile çalış

Diğer OpenAI modelleri