TPU v2
Questo documento descrive l'architettura e le configurazioni supportate Cloud TPU v2.
Architettura di sistema
I dettagli dell'architettura e le caratteristiche delle prestazioni di TPU v2 sono disponibili in Un supercomputer specifico per il dominio per l'addestramento delle reti neurali profonde Reti.
Configurazioni
Un pod TPU v2 è composto da 512 chip interconnessi con componenti
link ad alta velocità. Per creare una sezione di pod TPU v2, utilizza --accelerator-type
nel comando di creazione TPU (gcloud compute tpus tpu-vm
). Devi specificare
il tipo di acceleratore specificando la versione di TPU e il numero di core TPU. Per
ad esempio, per una singola TPU v2, utilizza --accelerator-type=v2-8
. Per una sezione di pod v2
con 128 TensorCore, usa --accelerator-type=v2-128
.
Il comando seguente mostra come creare una sezione di pod di TPU v2 con 128 TensorCore:
$ gcloud compute tpus tpu-vm create tpu-name \
--zone=zone \
--accelerator-type=v2-128 \
--version=tpu-vm-tf-2.17.0-pjrt
Per ulteriori informazioni sulla gestione delle TPU, consulta Gestire TPU. Per ulteriori informazioni sul sistema TPU per l'architettura di Cloud TPU, consulta dell'architettura.
Nella tabella seguente sono elencati i tipi di TPU v2 supportati:
Versione TPU | Data di 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) |