Créer une instance TPU

Ce document explique comment créer une instance de machine virtuelle (VM) TPU v6e (Trillium), s'y connecter et y exécuter un calcul. L'instance TPU que vous créez dans ce guide de démarrage rapide utilise l'option de consommation par défaut, à la demande.

Avant de commencer

  1. Connectez-vous à votre compte Google Cloud . Si vous débutez sur Google Cloud, créez un compte pour évaluer les performances de nos produits en conditions réelles. Les nouveaux clients bénéficient également de 300 $ de crédits sans frais pour exécuter, tester et déployer des charges de travail.

  2. Installez la Google Cloud CLI.

  3. Si vous utilisez un fournisseur d'identité (IdP) externe, vous devez d'abord vous connecter à la gcloud CLI avec votre identité fédérée.

  4. Pour initialiser la gcloud CLI, exécutez la commande suivante : 

    gcloud init

  5. Créez ou sélectionnez un projet Google Cloud .

    Rôles requis pour sélectionner ou créer un projet

    • Sélectionnez un projet : la sélection d'un projet ne nécessite pas de rôle IAM spécifique. Vous pouvez sélectionner n'importe quel projet pour lequel un rôle vous a été attribué.
    • Créer un projet : pour créer un projet, vous devez disposer du rôle Créateur de projet (roles/resourcemanager.projectCreator), qui contient l'autorisation resourcemanager.projects.create. Découvrez comment attribuer des rôles.

    • Créez un projet Google Cloud  :

      gcloud projects create PROJECT_ID

      Remplacez PROJECT_ID par le nom du projet Google Cloud que vous créez.

    • Sélectionnez le projet Google Cloud que vous avez créé :

      gcloud config set project PROJECT_ID

      Remplacez PROJECT_ID par le nom de votre projet Google Cloud .

  6. Si vous utilisez un projet existant pour ce guide, vérifiez que vous disposez des autorisations nécessaires pour suivre les instructions. Si vous avez créé un projet, vous disposez déjà des autorisations requises.

  7. Vérifiez que la facturation est activée pour votre projet Google Cloud .

  8. Activez l'API Compute Engine :

    Rôles requis pour activer les API

    Pour activer les API, vous avez besoin du rôle IAM Administrateur Service Usage (roles/serviceusage.serviceUsageAdmin), qui contient l'autorisation serviceusage.services.enable. Découvrez comment attribuer des rôles. 

    gcloud services enable compute.googleapis.com

  9. Installez la Google Cloud CLI.

  10. Si vous utilisez un fournisseur d'identité (IdP) externe, vous devez d'abord vous connecter à la gcloud CLI avec votre identité fédérée.

  11. Pour initialiser la gcloud CLI, exécutez la commande suivante : 

    gcloud init

  12. Créez ou sélectionnez un projet Google Cloud .

    Rôles requis pour sélectionner ou créer un projet

    • Sélectionnez un projet : la sélection d'un projet ne nécessite pas de rôle IAM spécifique. Vous pouvez sélectionner n'importe quel projet pour lequel un rôle vous a été attribué.
    • Créer un projet : pour créer un projet, vous devez disposer du rôle Créateur de projet (roles/resourcemanager.projectCreator), qui contient l'autorisation resourcemanager.projects.create. Découvrez comment attribuer des rôles.

    • Créez un projet Google Cloud  :

      gcloud projects create PROJECT_ID

      Remplacez PROJECT_ID par le nom du projet Google Cloud que vous créez.

    • Sélectionnez le projet Google Cloud que vous avez créé :

      gcloud config set project PROJECT_ID

      Remplacez PROJECT_ID par le nom de votre projet Google Cloud .

  13. Si vous utilisez un projet existant pour ce guide, vérifiez que vous disposez des autorisations nécessaires pour suivre les instructions. Si vous avez créé un projet, vous disposez déjà des autorisations requises.

  14. Vérifiez que la facturation est activée pour votre projet Google Cloud .

  15. Activez l'API Compute Engine :

    Rôles requis pour activer les API

    Pour activer les API, vous avez besoin du rôle IAM Administrateur Service Usage (roles/serviceusage.serviceUsageAdmin), qui contient l'autorisation serviceusage.services.enable. Découvrez comment attribuer des rôles. 

    gcloud services enable compute.googleapis.com
    16.

Rôles requis

Pour obtenir les autorisations nécessaires pour créer une instance de TPU, vous y connecter à l'aide de SSH et exécuter des commandes, demandez à votre administrateur de vous accorder les rôles IAM suivants sur votre projet :

Pour en savoir plus sur l'attribution de rôles, consultez la page Gérer l'accès aux projets, aux dossiers et aux organisations.

Vous pouvez également obtenir les autorisations requises avec des rôles personnalisés ou d'autres rôles prédéfinis.

Créer une instance TPU v6e (Trillium)

Exécutez les commandes suivantes dans votre terminal local ou à l'aide de Cloud Shell.

  1. Pour créer une instance TPU v6e, exécutez la commande gcloud compute instances create suivante :

    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. Vous pouvez également vérifier que l'instance est en cours d'exécution à l'aide de la commande gcloud compute instances describe :

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

    Dans le résultat, vérifiez le champ status pour savoir si l'instance est en cours d'exécution ou toujours en attente.

Se connecter à l'instance TPU

Vous pouvez vous connecter à une instance de TPU à l'aide de la console Google Cloud ou de la gcloud CLI.

Pour vous connecter à la VM à l'aide de SSH, exécutez la commande gcloud compute ssh suivante :

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

Installer JAX

Installez le framework JAX en exécutant la commande suivante sur votre instance TPU :

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

Exécuter une vérification du système

Vérifiez que JAX peut accéder au TPU et exécuter des opérations de base en exécutant les commandes suivantes sur votre instance de TPU :

  1. Démarrez l'interpréteur Python 3 :

    python3

  2. Importez le module JAX :

    >>> import jax

  3. Affichez le nombre de cœurs de TPU disponibles :

    >>> jax.device_count()

    Le résultat indique le nombre de cœurs de TPU. Le nombre de cœurs dépend de la version de TPU que vous utilisez. Pour en savoir plus, consultez Versions de TPU.

Effectuer un calcul

  1. Exécutez la commande suivante pour additionner deux nombres :

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

    La sortie devrait ressembler à ce qui suit :

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

  2. Quittez l'interpréteur Python :

    >>> 
exit()

Effectuer un nettoyage

Pour éviter que les ressources utilisées dans cette démonstration soient facturées sur votre compte Google Cloud , supprimez le projet Google Cloud qui les contient.

Pour éviter que les ressources utilisées sur cette page ne soient facturées sur votre compte Google Cloud , supprimez l'instance de TPU à l'aide de la commande gcloud compute instances delete :

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

Étapes suivantes