TPU v2

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

Architettura di sistema

I dettagli dell'architettura e le caratteristiche delle prestazioni di TPU v2 sono disponibili in A Domain Specific Supercomputer for Training Deep Neural Networks.

Configurazioni

Una sezione TPU v2 è composta da 512 chip interconnessi con link ad alta velocità riconfigurabili. Per creare una sezione TPU v2, utilizza il flag --accelerator-type nel comando di creazione TPU (gcloud compute tpus tpu-vm). Specifica il tipo di acceleratore specificando la versione 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