Créer une instance de TPU

Créer une instance de TPU

Ce document explique comment créer une instance de machine virtuelle (VM) TPU v6e (Trillium), vous y connecter et y exécuter un calcul. L'instance de 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

Connectez-vous à votre Google Cloud compte. Si vous n'avez jamais utilisé Google Cloud, créez un compte pour évaluer les performances de nos produits dans des scénarios réels. 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.

Installez la Google Cloud CLI.

Remarque : Si vous avez déjà installé la gcloud CLI, assurez-vous de disposer de la dernière version en exécutant gcloud components update.

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.

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

gcloud init

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

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

• Sélectionner 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 avez besoin du rôle Créateur de projet (roles/resourcemanager.projectCreator), qui contient l'autorisation resourcemanager.projects.create. Découvrez comment attribuer des rôles.

Remarque : Si vous ne comptez pas conserver les ressources créées dans cette procédure, créez un projet au lieu d'en sélectionner un existant. Après avoir suivi ces étapes, vous pouvez supprimer le projet. Cela entraîne la suppression de toutes les ressources qui lui sont associées.

• Créez un Google Cloud projet :

  gcloud projects create PROJECT_ID

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

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

  gcloud config set project PROJECT_ID

  Remplacez PROJECT_ID par le nom de votre Google Cloud projet.

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 nouveau projet, vous disposez déjà des autorisations requises.

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

Activez l'API Compute Engine :

Rôles requis pour activer les API

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

gcloud services enable compute.googleapis.com

Installez la Google Cloud CLI.

Remarque : Si vous avez déjà installé la gcloud CLI, assurez-vous de disposer de la dernière version en exécutant gcloud components update.

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.

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

gcloud init

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

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

• Sélectionner 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 avez besoin du rôle Créateur de projet (roles/resourcemanager.projectCreator), qui contient l'autorisation resourcemanager.projects.create. Découvrez comment attribuer des rôles.

Remarque : Si vous ne comptez pas conserver les ressources créées dans cette procédure, créez un projet au lieu d'en sélectionner un existant. Après avoir suivi ces étapes, vous pouvez supprimer le projet. Cela entraîne la suppression de toutes les ressources qui lui sont associées.

• Créez un Google Cloud projet :

  gcloud projects create PROJECT_ID

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

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

  gcloud config set project PROJECT_ID

  Remplacez PROJECT_ID par le nom de votre Google Cloud projet.

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 nouveau projet, vous disposez déjà des autorisations requises.

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

Activez l'API Compute Engine :

Rôles requis pour activer les API

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

gcloud services enable compute.googleapis.com

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 :

• Administrateur d'instances Compute (v1) (roles/compute.instanceAdmin.v1)
• Utilisateur du compte de service (roles/iam.serviceAccountUser)
• Administrateur Service Usage (roles/serviceusage.serviceUsageAdmin)

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 de TPU v6e (Trillium)

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

1. Pour créer une instance de 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 gcloud compute instances describe commande :

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

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

Se connecter à l'instance de TPU

Vous pouvez vous connecter à une instance de TPU à l'aide de la Google Cloud console 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 de TPU :

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

Remarque : Si cette commande échoue, vous devrez peut-être exécuter sudo apt update et sudo apt install python3-pip.

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 affiche 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()
   

Libérer de l'espace

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

Si vous souhaitez conserver votre projet, vous pouvez supprimer uniquement la VM en utilisant la gcloud compute instances delete commande :

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

Étape suivante

• En savoir plus sur les ressources Cloud TPU dans Compute Engine
• Créer une instance de Cloud TPU à l'aide d'autres options de consommation
• Créer des instances de Cloud TPU avec des MIG