Dukungan Dataflow untuk TPU

Google Cloud TPU adalah akselerator AI yang dirancang khusus dan dibuat oleh Google yang dioptimalkan untuk melatih dan menggunakan model AI berskala besar. Cloud TPU dirancang untuk menskalakan berbagai workload AI secara hemat biaya dan memberikan fleksibilitas untuk mempercepat workload inferensi pada framework AI, termasuk PyTorch, JAX, dan TensorFlow. Untuk mengetahui detail selengkapnya tentang TPU, lihat Pengantar Google Cloud TPU.

Prasyarat untuk menggunakan TPU di Dataflow

  • Project Google Cloud Anda harus disetujui untuk menggunakan penawaran GA ini.

Batasan

Penawaran ini tunduk pada batasan berikut:

  • Hanya akselerator TPU host tunggal yang didukung: Penawaran TPU Dataflow hanya mendukung konfigurasi TPU host tunggal di mana setiap pekerja Dataflow mengelola satu atau beberapa perangkat TPU yang tidak saling terhubung dengan TPU yang dikelola oleh pekerja lain.
  • Hanya kumpulan pekerja TPU homogen yang didukung: Fitur seperti Dataflow right fitting dan Dataflow Prime tidak mendukung beban kerja TPU.

Harga

Tugas Dataflow yang menggunakan TPU ditagih untuk jam-chip TPU pekerja yang digunakan dan tidak ditagih untuk CPU dan memori pekerja. Untuk mengetahui informasi selengkapnya, lihat halaman harga Dataflow.

Ketersediaan

Akselerator TPU dan region pemrosesan berikut tersedia.

Akselerator TPU yang didukung

Kombinasi akselerator TPU yang didukung diidentifikasi oleh tuple (jenis TPU, topologi TPU).

  • Jenis TPU mengacu pada model perangkat TPU.
  • Topologi TPU mengacu pada jumlah dan pengaturan fisik chip TPU dalam slice.

Untuk mengonfigurasi jenis dan topologi TPU untuk pekerja Dataflow, gunakan opsi pipeline worker_accelerator yang diformat sebagai type:TPU_TYPE;topology:TPU_TOPOLOGY.

Konfigurasi TPU berikut didukung dengan Dataflow:

Jenis TPU Topologi Wajib worker_machine_type
tpu-v5-lite-podslice 1x1 ct5lp-hightpu-1t
tpu-v5-lite-podslice 2x2 ct5lp-hightpu-4t
tpu-v5-lite-podslice 2x4 ct5lp-hightpu-8t
tpu-v6e-slice 1x1 ct6e-standard-1t
tpu-v6e-slice 2x2 ct6e-standard-4t
tpu-v6e-slice 2x4 ct6e-standard-8t
tpu-v5p-slice 2x2x1 ct5p-hightpu-4t

Region

Untuk mengetahui informasi tentang region dan zona yang tersedia untuk TPU, lihat Region dan zona TPU di dokumentasi Cloud TPU.

Langkah berikutnya