TPU v2

Perhatian: TPU v2 hanya didukung menggunakan Cloud TPU API, yang tidak lagi dalam pengembangan aktif. Cloud TPU API hanya akan menerima perbaikan bug dan update keamanan. Anda harus bermigrasi ke versi TPU yang lebih baru dan menggunakan Compute Engine atau Google Kubernetes Engine untuk mengelola resource TPU. Untuk mengetahui informasi selengkapnya, lihat Resource TPU di Compute Engine dan Tentang TPU di GKE.

Dokumen ini menjelaskan arsitektur dan konfigurasi Cloud TPU v2 yang didukung.

Arsitektur sistem

Detail arsitektur dan karakteristik performa TPU v2 tersedia di Superkomputer Khusus Domain untuk Melatih Jaringan Neural Dalam.

Konfigurasi

Pod TPU v2 terdiri dari hingga 256 chip (512 TensorCore) yang saling terhubung dengan link berkecepatan tinggi yang dapat dikonfigurasi ulang. Untuk membuat slice TPU v2, gunakan flag --accelerator-type dalam perintah pembuatan TPU (gcloud compute tpus tpu-vm). Anda menentukan jenis akselerator dengan menentukan versi TPU dan jumlah core TPU. Misalnya, untuk satu TPU v2, gunakan --accelerator-type=v2-8. Untuk slice v2 dengan 128 TensorCore, gunakan --accelerator-type=v2-128.

Perintah berikut menunjukkan cara membuat slice TPU v2 dengan 128 TensorCore:

  $ gcloud compute tpus tpu-vm create tpu-name \
    --zone=us-central1-a \
    --accelerator-type=v2-128 \
    --version=tpu-ubuntu2204-base

Untuk mengetahui informasi selengkapnya tentang cara mengelola TPU, lihat Mengelola TPU. Untuk mengetahui informasi selengkapnya tentang arsitektur sistem TPU Cloud TPU, lihat Arsitektur sistem.

Tabel berikut mencantumkan jenis TPU v2 yang didukung:

Versi TPU	Dukungan berakhir
v2-8	(Tanggal akhir belum ditetapkan)
v2-32	(Tanggal akhir belum ditetapkan)
v2-128	(Tanggal akhir belum ditetapkan)
v2-256	(Tanggal akhir belum ditetapkan)
v2-512	(Tanggal akhir belum ditetapkan)