TPU-Softwareversionen

In diesem Dokument finden Sie eine Anleitung zur Auswahl der geeigneten TPU-Softwareversion beim Erstellen von Cloud TPUs.

Wenn Sie TPU-Ressourcen erstellen, geben Sie die auch als Laufzeitversion bezeichnete Softwareversion an. Sie bezieht sich auf die Softwareumgebung, die auf Ihrer TPU-VM vorinstalliert ist. Dazu gehören das Ubuntu-Betriebssystem, Docker und andere Software, die zum Ausführen Ihres Codes auf TPUs erforderlich ist.

Wenn Sie die Google Cloud CLI verwenden, geben Sie die TPU-Softwareversion mit dem Parameter --version oder --runtime-version an. Wenn Sie die Google Cloud Console verwenden, wählen Sie eine TPU-Softwareversion aus der Liste Version der TPU-Software aus.

PyTorch und JAX

Verwenden Sie die folgenden gängigen TPU-Softwareversionen für PyTorch und JAX und installieren Sie dann das gewünschte Framework.