✅ — TPU v6e Can Run Gemma 2 2B
The TPU v6e packs 256GB VRAM. Gemma 2 2B needs ~1.1GB. 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 2B
- Parameters
- 2B
- VRAM (Q4)
- ~1.1 GB
- Context
- N/A
Estimated Performance
3375 tok/s
Estimated inference speed (Q4 quantization)