✅ — TPU v6e Can Run Gemma 2 9B
The TPU v6e packs 256GB VRAM. Gemma 2 9B needs ~5GB. You're good to go.
✅ Plenty of headroom
Hardware Specs
- GPU
- TPU v6e
- VRAM
- 256 GB
- Bandwidth
- 4500 GB/s
- MSRP
- $0
Model Requirements
- Model
- Gemma 2 9B
- Parameters
- 9B
- VRAM (Q4)
- ~5 GB
- Context
- N/A
Estimated Performance
750 tok/s
Estimated inference speed (Q4 quantization)