Gemma 2 27B IT kaç GB VRAM ister?

Gemma 2 27B IT, 46 katmanda 27,20B parametreye sahiptir. Grouped-query attention kullanır: 32 query head, 16 key-value head'i paylaşır; bu yüzden KV cache'i tam multi-head attention'a göre 2× küçüktür. En fazla 8.192 token context alır; bu uzunlukta tek bir kullanıcı, ağırlıkların üstüne FP16'da 2,88 GiB KV cache ister. Hugging Face deposu erişim onayı gerektirdiği için LLMScale mimarisini yerleşik veritabanında taşır.

BF16 · 8k context · tek kullanıcı
56,79 GiB

50,66 GiB ağırlık, 2,88 GiB KV cache ve 3,25 GiB vLLM'in ek yükü. Aynı iş yükü INT4'te 21,19 GiB ister.

  • 46 katman
  • hidden 4608
  • 32/16 heads
  • head_dim 128
  • 8k maks context

Hassasiyet ve context'e göre Gemma 2 27B IT belleği

Hassasiyet8k context
BF1656,79 GiBKV cache 2,88 GiB
FP832,55 GiBKV cache 2,88 GiB
INT421,19 GiBKV cache 2,88 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ığmaz1
RTX 5090sığmazsığmaz3
RTX 6000 Adasığmaz48
H100 SXM 80GB61418
RTX PRO 6000 Blackwell111923
AMD Ryzen AI Max+ 395 (128GB)112024
H200 141GB253438
B200 180GB374549

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

Gemma 2 27B IT ince ayarı

YöntemBellekEn küçük GPU
Tam ince ayar407,1 GiBbirden çok GPU, paylaştırılmış
LoRA (rank 16, attention)52,78 GiBNVIDIA Jetson AGX Orin (64GB)
QLoRA (rank 16, attention)15,18 GiBRTX 4060 Ti 16GB

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. Gemma 2 27B IT için bu, FP16'da token başına 368,0 KiB eder. 46 katmanının 23 tanesi son 4.096 token'lık kayan pencereye bakar ve bundan fazlasını tutmaz; pencereden sonra eklenen her token, diğer 23 katmanın payı olan 184,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.

Gemma 2 27B IT ile çalış

Diğer Gemma modelleri