Calculadora de GPU

¿Qué GPU para mi modelo?

Elija un modelo, su precisión, su contexto y su concurrencia. Vea la memoria GPU que necesita, línea por línea, y las GPU que lo admiten.

Calculadora

8.03 mil millones de parámetros · Llama 3.1 Community License

Necesita 19.5 GiB de memoria GPU. Confianza alta

Memoria GPU, línea por línea
Pesos del modelo8.03B parameters × 2 bytes (16-bit (FP16 or BF16))15.0 GiB
Caché KV2 × 32 layers × 8 KV heads × 128 head dim × 2 bytes × 4,096 tokens × 1 concurrent0.5 GiB
Margen adicional20 % of the figures above (activations, allocator, quantization scales)3.1 GiB
Runtime CUDA1 GiB per GPU (CUDA context and library workspaces)1.0 GiB
Total19.5 GiB
Qué GPU lo admiten
Memoria GPU1 GPU2 GPU4 GPU8 GPU
24 GB por GPUCabeCabe en 1Cabe en 1Cabe en 1
48 GB por GPUCabeCabe en 1Cabe en 1Cabe en 1
80 GB por GPUCabeCabe en 1Cabe en 1Cabe en 1

Cada solicitud puede usar el contexto completo. Caché KV en 16 bits.

La memoria se calcula a partir de los datos publicados de cada modelo: pesos (parámetros × bytes por parámetro), caché KV (capas × cabezas KV × dimensión de cabeza × contexto × solicitudes), un margen del 20 % y 1 GiB de runtime CUDA. Es una estimación, no una medición: los frameworks reservan la memoria de forma distinta, la velocidad no se estima y los modelos de imagen solo cuentan sus pesos.

Encuentre en la consola una máquina que lo admita