Créer une instance TPU

Ce document vous explique comment créer une instance de machine virtuelle (VM) TPU, vous 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 à la demande, qui est l'option par défaut.

Avant de commencer

  1. Installez la Google Cloud CLI.

  2. Configurez la gcloud CLI afin d'utiliser votre identité fédérée.

    Pour en savoir plus, consultez Se connecter à la gcloud CLI avec votre identité fédérée.

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

    gcloud init

  4. Créez ou sélectionnez un projet Cloud de Confiance .

    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 Cloud de Confiance  :

      gcloud projects create PROJECT_ID

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

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

      gcloud config set project PROJECT_ID

      Remplacez PROJECT_ID par le nom de votre projet Cloud de Confiance .

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

  6. Vérifiez que la facturation est activée pour votre projet Cloud de Confiance .

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

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

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

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

    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. 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=us-east5-b

    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 Cloud de Confiance 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 --project=PROJECT_ID --zone=us-east5-b

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 Cloud de Confiance , supprimez le projet Cloud de Confiance qui les contient.

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

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

Étapes suivantes