Can I run gpt-neox-20b on an NVIDIA H100?
Yes
Its published weights (41.3 GB) fit in the 72.0 GB an NVIDIA H100 can use, with room for ≈ 2K tokens of context.
hf download EleutherAI/gpt-neox-20bFiles and context on NVIDIA H100
Each file of gpt-neox-20b (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 NVIDIA H100 | Context (16-bit) | Context (8-bit) |
|---|---|---|---|---|
| As published | 41.3 GB | Fits | 2K | 2K |
| 4-bit (estimate) | 11.7 GB | Fits | 2K | 2K |