Skip to main content
Computed

DeepSeek-R1-Distill-Qwen 1.5B on an A100 40GB (SXM)

Yes — the unquantized weights fit with room for 128k tokens of context.

Verdict

Fits at FP16

36.8 GiB usable of 40 GB

Weights (Q4_K_M)

1.0 GiB

3.3 GiB at FP16 · 1.8B params · 6 of 6 quant rows are measured files

KV cache

28.0 KiB per token at FP16

28 layers × 2 KV heads × 128 dimensions

Speed ceiling

410 tok/s

1555 GB/s ÷ bytes read per token

Every quant, against this card

Weight bytes are the size of the real published file wherever one exists — 6 of these6 rows are measured from bartowski/DeepSeek-R1-Distill-Qwen-1.5B-GGUF, the rest computed from the parameter count. Max context is what the KV cache can grow to in whatever memory the weights leave behind, capped at the 128k tokens this model was trained to address.

36.8 GiB usableFP16 / BF16 · 3.3 GiBQ8_0 · 1.8 GiBQ6_K · 1.4 GiBQ5_K_M · 1.2 GiBQ4_K_M · 1.0 GiBQ3_K_M · 882 MiB
Bars are the weight bytes at each precision; the dashed line is the usable memory of an A100 40GB (SXM). Drawn from the computed byte counts, not sketched.
PrecisionBits/weightWeightsSourceFitsMax contextCeiling
FP16 / BF1616.003.3 GiBmeasured fileyes131,072 (model cap)410 tok/s
Q8_08.531.8 GiBmeasured fileyes131,072 (model cap)730 tok/s
Q6_K6.591.4 GiBmeasured fileyes131,072 (model cap)915 tok/s
Q5_K_M5.791.2 GiBmeasured fileyes131,072 (model cap)1023 tok/s
Q4_K_M5.031.0 GiBmeasured fileyes131,072 (model cap)1150 tok/s
Q3_K_M4.16882 MiBmeasured fileyes131,072 (model cap)1341 tok/s

What the context actually costs

The KV cache is 2 · layers · kv_heads · head_dim bytes per token per element — 2 · 28 · 2 · 128 · 2 B = 28.0 KiB. It depends on the key/value head count, not the attention-head count and not the parameter count. This model shares 2 KV heads across 12 query heads, which divides the cache by 6 against multi-head attention.

ContextKV cache (FP16)KV cache (8-bit)Plus Q4_K_M weights
4,096112 MiB56 MiB1.1 GiB
8,192224 MiB112 MiB1.3 GiB
32,768896 MiB448 MiB1.9 GiB
131,0723.5 GiB1.8 GiB4.5 GiB

The speed ceiling, and where it comes from

Generating one token reads every active weight from memory once. At FP16 / BF16 that is 3.3 GiB, plus a pass over the KV cache. An A100 40GB (SXM) moves 1555 GB/s, so the arithmetic ceiling is 410 tok/s. Treat it as a bound, not an estimate: attention overhead, kernel launches and imperfect memory access keep real runtimes at roughly 60–80% of it, and nothing pushes past it.

Where these numbers come from

The model

Parameters
1,777,088,000
Layers
28
Attention / KV heads
12 / 2
Head dimension
128
Trained context
131,072
Checkpoint as published
3.3 GiB

Read from deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B. The parameter count is the Hub's own total over the tensor shapes, not a figure taken from the model's name.

The accelerator

Memory
40 GB HBM2
Bandwidth
1555 GB/s
Assumed usable
92% → 36.8 GiB

Capacity and bandwidth from the vendor's specification. The usable fraction is an assumption, not a spec: a driver context, compute workspace and any attached display come out of the same pool before a weight is loaded.

Questions this pairing answers

How much VRAM does DeepSeek-R1-Distill-Qwen 1.5B need?

3.3 GiB for the weights at FP16 — 1.8B parameters at two bytes each — and 1.0 GiB at Q4_K_M. The KV cache is on top of that and is not a fixed number: this model spends 28.0 KiB per token of context, so 8,192 tokens costs a further 224 MiB. An A100 40GB (SXM) makes 36.8 GiB of its 40 GB available on the assumption below.

Can an A100 40GB (SXM) run DeepSeek-R1-Distill-Qwen 1.5B?

Yes — the unquantized weights fit with room for 128k tokens of context. That is the weights and the KV cache together, against 36.8 GiB of usable memory.

How fast will DeepSeek-R1-Distill-Qwen 1.5B run on an A100 40GB (SXM)?

No faster than 410 tokens/second at FP16 / BF16, and in practice below it. Decoding is memory-bound: every token reads all 3.3 GiB of weights plus the cache, and this card moves 1555 GB/s. That division is the ceiling — no kernel, runtime or driver beats it, and a real runtime typically reaches 60–80% of it.

Why does the context length change how much memory DeepSeek-R1-Distill-Qwen 1.5B needs?

Because the KV cache holds one key and one value vector per token, per layer, for the whole conversation, and it is allocated separately from the weights. This model has 28 layers and 2 key/value heads of 128 dimensions, shared across 12 query heads — grouped-query attention, which divides the cache by 6. That works out at 28.0 KiB per token. Parameter count tells you nothing about this number.

The same model on a different card

A different model on the same card

All 62 models on A100 40GB (SXM) →

Method and limits. Weight bytes are the byte size of the real published file wherever one exists, and the model's exact parameter count times the published llama.cpp bits-per-weight where it does not. That distinction is on every row above and it matters at both ends: a sub-1B model's Q4_K_M file runs a third larger than the nominal figure because k-quants keep its embedding tables at higher precision, and an already-4-bit release cannot be quantized upward at all. The KV cache is 2 · layers · kv_heads · head_dim · bytes per token, summed over layers with each sliding-window layer capped at its window. The speed figure is a roofline bound, not a benchmark: bandwidth divided by bytes read per token, which no runtime exceeds and every runtime falls short of. The usable fraction of card memory is an assumption: 92% here, which is what this lane assumes for a dedicated card — the other class assumes 75%, so it is not one number applied to every device. That is the only assumed input on this page; every other figure is computed from the model config and the card's published specification. Nothing on this page is written by a language model. Architecture from the model's published config, fetched 2026-08-07.