TPU-Instanz erstellen

In diesem Dokument wird beschrieben, wie Sie eine TPU v6e-VM-Instanz (Trillium) erstellen, eine Verbindung zu ihr herstellen und eine Berechnung darauf ausführen. Für die TPU-Instanz, die Sie in dieser Kurzanleitung erstellen, wird die Standardoption für die Nutzung „On-Demand“ verwendet.

Hinweis

  1. Installieren Sie die Google Cloud CLI.

  2. Konfigurieren Sie die gcloud CLI für die Verwendung Ihrer föderierten Identität.

    Weitere Informationen finden Sie unter Mit Ihrer föderierten Identität in der gcloud CLI anmelden.

  3. Führen Sie den folgenden Befehl aus, um die gcloud CLI zu initialisieren: 

    gcloud init

  4. Cloud de Confiance Projekt erstellen oder auswählen

    Erforderliche Rollen zum Auswählen oder Erstellen eines Projekts

    • Projekt auswählen: Für die Auswahl eines Projekts ist keine bestimmte IAM-Rolle erforderlich. Sie können jedes Projekt auswählen, für das Ihnen eine Rolle zugewiesen wurde.
    • Projekt erstellen: Zum Erstellen eines Projekts benötigen Sie die Rolle „Project Creator“ (roles/resourcemanager.projectCreator), die die Berechtigung resourcemanager.projects.create enthält. Informationen zum Zuweisen von Rollen

    • So erstellen Sie ein Cloud de Confiance Projekt:

      gcloud projects create PROJECT_ID

      Ersetzen Sie PROJECT_ID durch einen Namen für das Cloud de Confiance Projekt, das Sie erstellen.

    • Wählen Sie das von Ihnen erstellte Cloud de Confiance Projekt aus:

      gcloud config set project PROJECT_ID

      Ersetzen Sie PROJECT_ID durch Ihren Cloud de Confiance Projektnamen.

  5. Wenn Sie für diese Anleitung ein vorhandenes Projekt verwenden, prüfen Sie, ob Sie die erforderlichen Berechtigungen haben. Wenn Sie ein neues Projekt erstellt haben, haben Sie bereits die erforderlichen Berechtigungen.

  6. Prüfen Sie, ob die Abrechnung für Ihr Cloud de Confiance Projekt aktiviert ist.

  7. Aktivieren Sie die Compute Engine API, falls sie noch nicht aktiviert ist:

    Rollen, die zum Aktivieren von APIs erforderlich sind

    Zum Aktivieren von APIs benötigen Sie die Berechtigung serviceusage.services.enable. Wenn Sie das Projekt erstellt haben, haben Sie diese Berechtigung wahrscheinlich bereits über die Rolle „Inhaber“ (roles/owner). Andernfalls können Sie diese Berechtigung über die Rolle „Service Usage-Administrator“ (roles/serviceusage.serviceUsageAdmin) erhalten. Informationen zum Zuweisen von Rollen 

    gcloud services enable compute.googleapis.com

Erforderliche Rollen

Bitten Sie Ihren Administrator, Ihnen die folgenden IAM-Rollen für Ihr Projekt zuzuweisen, um die Berechtigungen zu erhalten, die Sie zum Erstellen einer TPU-Instanz, zum Herstellen einer SSH-Verbindung und zum Ausführen von Befehlen benötigen:

Weitere Informationen zum Zuweisen von Rollen finden Sie unter Zugriff auf Projekte, Ordner und Organisationen verwalten.

Sie können die erforderlichen Berechtigungen auch über benutzerdefinierte Rollen oder andere vordefinierte Rollen erhalten.

TPU v6e-Instanz (Trillium) erstellen

Führen Sie die folgenden Befehle in Ihrem lokalen Terminal oder mit Cloud Shell aus.

  1. Führen Sie den folgenden gcloud compute instances create-Befehl aus, um eine TPU v6e-Instanz zu erstellen:

    gcloud compute instances create quickstart-tpu \
    --machine-type=ct6e-standard-4t \
    --zone=europe-west4-a \
    --image-project=ubuntu-os-accelerator-images \
    --image-family=ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e \
    --maintenance-policy=TERMINATE

  2. Optional können Sie mit dem Befehl gcloud compute instances describe prüfen, ob die Instanz ausgeführt wird:

    gcloud compute instances describe quickstart-tpu --zone=europe-west4-a

    Prüfen Sie in der Ausgabe das Feld status, um festzustellen, ob die Instanz ausgeführt wird oder noch aussteht.

Verbindung zur TPU-Instanz herstellen

Sie können über die Cloud de Confiance Console oder die gcloud CLI eine Verbindung zu einer TPU-Instanz herstellen.

Stellen Sie mit SSH eine Verbindung zur VM her, indem Sie den Befehl gcloud compute ssh ausführen:

gcloud compute ssh quickstart-tpu --zone=europe-west4-a

JAX installieren

Installieren Sie das JAX-Framework, indem Sie den folgenden Befehl auf Ihrer TPU-Instanz ausführen:

pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html

Systemprüfung durchführen

Prüfen Sie, ob JAX auf die TPU zugreifen und grundlegende Vorgänge ausführen kann, indem Sie die folgenden Befehle auf Ihrer TPU-Instanz ausführen:

  1. Starten Sie den Python 3-Interpreter:

    python3

  2. Importieren Sie das JAX-Modul:

    >>> import jax

  3. Prüfen Sie die Anzahl der verfügbaren TPU-Kerne:

    >>> jax.device_count()

    In der Ausgabe wird die Anzahl der TPU-Kerne angezeigt. Die Anzahl der Kerne hängt von der verwendeten TPU-Version ab. Weitere Informationen finden Sie unter TPU-Versionen.

Berechnung durchführen

  1. Führen Sie den folgenden Befehl aus, um zwei Zahlen zu addieren:

    >>> jax.numpy.add(1, 1)

    Die Ausgabe sollte in etwa so aussehen:

    >>> Array(2, dtype=int32, weak_type=True)

  2. Beenden Sie den Python-Interpreter:

    >>> 
exit()

Bereinigen

Löschen Sie das Projekt von Cloud de Confiance zusammen mit den Ressourcen, damit Ihrem Konto von Cloud de Confiance die auf dieser Seite verwendeten Ressourcen nicht in Rechnung gestellt werden.

Wenn Sie Ihr Projekt behalten möchten, können Sie alternativ nur die VM mit dem gcloud compute instances delete-Befehl löschen:

gcloud compute instances delete quickstart-tpu --zone=europe-west4-a

Nächste Schritte