TPU v2

Attenzione: TPU v2 è supportato solo utilizzando l'API Cloud TPU, che non è più in fase di sviluppo attivo. L'API Cloud TPU riceverà solo correzioni di bug e aggiornamenti della sicurezza. Devi eseguire la migrazione a una versione TPU più recente e utilizzare Compute Engine o Google Kubernetes Engine per gestire le risorse TPU. Per saperne di più, consulta Risorse TPU in Compute Engine e Informazioni sulle TPU in GKE.

Questo documento descrive l'architettura e le configurazioni supportate di Cloud TPU v2.

Architettura di sistema

I dettagli architettonici e le caratteristiche prestazionali di TPU v2 sono disponibili in Un supercomputer specifico per il dominio per l'addestramento di reti neurali profonde.

Configurazioni

Un pod TPU v2 è composto da un massimo di 256 chip (512 TensorCore) interconnessi con link ad alta velocità riconfigurabili. Per creare una sezione TPU v2, utilizza il flag --accelerator-type nel comando di creazione della TPU (gcloud compute tpus tpu-vm). Specifica il tipo di acceleratore indicando la versione della TPU e il numero di core TPU. Ad esempio, per una singola TPU v2, utilizza --accelerator-type=v2-8. Per una sezione v2 con 128 TensorCore, utilizza --accelerator-type=v2-128.

Il seguente comando mostra come creare una sezione TPU v2 con 128 TensorCore:

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

Per saperne di più sulla gestione delle TPU, consulta Gestisci le TPU. Per ulteriori informazioni sull'architettura di sistema di Cloud TPU, consulta Architettura di sistema.

Nella tabella seguente sono elencati i tipi di TPU v2 supportati:

Versione TPU	Fine del supporto
v2-8	(Data di fine non ancora impostata)
v2-32	(Data di fine non ancora impostata)
v2-128	(Data di fine non ancora impostata)
v2-256	(Data di fine non ancora impostata)
v2-512	(Data di fine non ancora impostata)