TPU v2

In diesem Dokument werden die Architektur und die unterstützten Konfigurationen von Cloud TPU v2 beschrieben.

Systemarchitektur

Details zur Architektur und zu den Leistungsmerkmalen von TPU v2 sind unter Domainspezifischer Supercomputer zum Trainieren neuronaler Deep-Learning-Netzwerke verfügbar.

Konfigurationen

Ein TPU v2-Speicher besteht aus 512 Chips, die über rekonfigurierbare Hochgeschwindigkeitsverbindungen miteinander verbunden sind. Verwenden Sie das Flag --accelerator-type im Befehl zum Erstellen von TPUs (gcloud compute tpus tpu-vm), um ein TPU v2-Speicher-Slice zu erstellen. Geben Sie den Accelerator-Typ an, indem Sie die TPU-Version und die Anzahl der TPU-Cores angeben. Verwenden Sie für eine einzelne v2-TPU beispielsweise --accelerator-type=v2-8. Verwenden Sie für einen v2-Speicherbereich mit 128 TensorCores --accelerator-type=v2-128.

Der folgende Befehl zeigt, wie Sie ein TPU-Stück der Version 2 mit 128 TensorCores erstellen:

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

Weitere Informationen zum Verwalten von TPUs finden Sie unter TPUs verwalten. Weitere Informationen zur TPU-Systemarchitektur von Cloud TPU finden Sie unter Systemarchitektur.

In der folgenden Tabelle sind die unterstützten TPU-Typen der Version 2 aufgeführt: