Can I run distilgpt2 on an RTX 3090?
Yes
Its published weights (353 MB) fit in the 21.6 GB an RTX 3090 can use, with room for ≈ 1K tokens of context.
hf download distilbert/distilgpt2Files and context on RTX 3090
Each file of distilgpt2 (and of GGUF versions the site has analyzed), and how much context fits next to it, with the context cache in 16-bit or 8-bit.
| File | Size | On RTX 3090 | Context (16-bit) | Context (8-bit) |
|---|---|---|---|---|
| As published | 353 MB | Fits | 1K | 1K |
| 4-bit (estimate) | 49.6 MB | Fits | 1K | 1K |