TPU v2

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: