TPU-Instanz erstellen

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

Hinweis

  1. Melden Sie sich in Ihrem Google Cloud Konto an. Wenn Sie noch kein Google CloudKonto haben, erstellen Sie ein Konto, um die Leistungsfähigkeit unserer Produkte in der Praxis sehen und bewerten zu können. Neukunden erhalten außerdem ein Guthaben von 300 $, um Arbeitslasten auszuführen, zu testen und bereitzustellen.

  2. Installieren Sie die Google Cloud CLI.

  3. Wenn Sie einen externen Identitätsanbieter (IdP) verwenden, müssen Sie sich zuerst mit Ihrer föderierten Identität in der gcloud CLI anmelden.

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

    gcloud init

  5. Erstellen Sie ein neues Projekt oder wählen Sie ein vorhandenes Projekt Google Cloud aus.

    Rollen, die zum Auswählen oder Erstellen eines Projekts erforderlich sind

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

    • Projekt erstellen: Google Cloud 

      gcloud projects create PROJECT_ID

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

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

      gcloud config set project PROJECT_ID

      Ersetzen Sie PROJECT_ID durch den Namen Ihres Google Cloud Projekts.

  6. 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.

  7. Prüfen Sie, ob für Ihr Google Cloud Projekt die Abrechnung aktiviert ist.

  8. Aktivieren Sie die Compute Engine API:

    Rollen, die zum Aktivieren von APIs erforderlich sind

    Zum Aktivieren von APIs benötigen Sie die IAM-Rolle „Service Usage-Administrator“ (roles/serviceusage.serviceUsageAdmin), die die Berechtigung serviceusage.services.enable enthält. Weitere Informationen zum Zuweisen von Rollen. 

    gcloud services enable compute.googleapis.com

  9. Installieren Sie die Google Cloud CLI.

  10. Wenn Sie einen externen Identitätsanbieter (IdP) verwenden, müssen Sie sich zuerst mit Ihrer föderierten Identität in der gcloud CLI anmelden.

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

    gcloud init

  12. Erstellen Sie ein neues Projekt oder wählen Sie ein vorhandenes Projekt Google Cloud aus.

    Rollen, die zum Auswählen oder Erstellen eines Projekts erforderlich sind

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

    • Projekt erstellen: Google Cloud 

      gcloud projects create PROJECT_ID

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

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

      gcloud config set project PROJECT_ID

      Ersetzen Sie PROJECT_ID durch den Namen Ihres Google Cloud Projekts.

  13. 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.

  14. Prüfen Sie, ob für Ihr Google Cloud Projekt die Abrechnung aktiviert ist.

  15. Aktivieren Sie die Compute Engine API:

    Rollen, die zum Aktivieren von APIs erforderlich sind

    Zum Aktivieren von APIs benötigen Sie die IAM-Rolle „Service Usage-Administrator“ (roles/serviceusage.serviceUsageAdmin), die die Berechtigung serviceusage.services.enable enthält. Weitere Informationen zum Zuweisen von Rollen. 

    gcloud services enable compute.googleapis.com
    16.

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 Verbindung zu ihr über SSH 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-Instanz erstellen

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

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

    gcloud compute instances create quickstart-tpu \
    --machine-type=ct6e-standard-4t \
    --zone=us-east5-b \
    --image-project=ubuntu-os-accelerator-images \
    --image-family=ubuntu-accelerator-2204-amd64-with-tpu-v6e

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

    gcloud compute instances describe quickstart-tpu --zone=us-east5-b

    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 Google Cloud console oder die gcloud CLI eine Verbindung zu einer TPU-Instanz herstellen.

Führen Sie den folgenden gcloud compute ssh Befehl aus, um über SSH eine Verbindung zur VM herzustellen:

gcloud compute ssh quickstart-tpu --project=PROJECT_ID --zone=us-east5-b

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

Systemüberprüfung ausfü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. Rufen Sie die Anzahl der verfügbaren TPU-Kerne auf:

    >>> jax.device_count()

    Die Ausgabe zeigt die Anzahl der TPU-Kerne. 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 zusammen mit den Ressourcen, damit Ihrem Google Cloud Konto von die auf dieser Seite verwendeten Ressourcen nicht in Rechnung gestellt werden. Google Cloud

Löschen Sie die TPU-Instanz mit dem gcloud compute instances delete Befehl, damit Ihrem Konto von die auf dieser Seite verwendeten Ressourcen nicht in Rechnung gestellt werden: Google Cloud 

gcloud compute instances delete quickstart-tpu --zone=us-east5-b

Nächste Schritte