Connecter un TPU à un réseau VPC partagé

Important :Ce guide explique comment configurer des Cloud TPU qui utilisent un réseau VPC partagé géré de manière centralisée. Ce document part du principe que vous connaissez les VPC et que vous avez créé un VPC partagé. Pour en savoir plus sur la création et la gestion de réseaux VPC partagés, consultez la page Créer et modifier des réseaux VPC partagés.

Configurer un projet hôte VPC

Vous devez accorder au compte de service TPU de votre projet de service les autorisations nécessaires pour gérer les ressources du projet hôte. Pour ce faire, utilisez le rôle "Agent dans le VPC partagé pour les nœuds TPU" (roles/tpu.xpnAgent). Exécutez les commandes gcloud suivantes pour accorder cette liaison de rôle.

gcloud projects add-iam-policy-binding host-project-id \
--member=serviceAccount:[e-mail] \
--role=roles/tpu.xpnAgent

Remarque :Vous trouverez le numéro de votre projet de service dans la section "Informations sur le projet" du tableau de bord de la console Google Cloud.

Créer une VM TPU connectée à un réseau VPC partagé

Déterminez d'abord les types et versions d'accélérateurs disponibles dans la zone.

gcloud compute tpus accelerator-types list --zone zone

gcloud compute tpus versions list --zone zone

Vous connectez une VM TPU à un réseau VPC partagé lorsque vous créez votre TPU. Spécifiez votre VPC partagé à l'aide de la balise --network:

gcloud compute tpus tpu-vm create tpu-name \
   --zone zone \
   --accelerator-type accelerator-type \
   --network projects/host-project-id/global/networks/host-network \
   --version tpu-image-version \
   --project your-service-project-id

Remarque :La balise --network doit être définie sur le nom de réseau complet. Exemple : projects/my-host-project-id/global/networks/my-network.

Vous pouvez vérifier que votre VM TPU est connectée à votre VPC partagé à l'aide de la commande gcloud describe:

$ gcloud compute tpus tpu-vm describe tpu-name --zone zone

La réponse inclut le réseau auquel votre VM TPU est associée:

acceleratorType: v3-8
apiVersion: V2
cidrBlock: 10.128.0.0/20
createTime: '2022-06-17T21:32:13.859274143Z'
health: HEALTHY
id: '0000000000000000000'
name: projects/my-project/locations/us-central1-b/nodes/my-tpu
networkConfig:
  enableExternalIps: true
  network: projects/my-project/global/networks/default
  subnetwork: projects/my-project/regions/us-central1/subnetworks/default
networkEndpoints:
- accessConfig:
    externalIp: 000.000.000.000
  ipAddress: 10.128.0.104
  port: 8470
runtimeVersion: tpu-vm-tf-2.8.0
schedulingConfig: {}
serviceAccount:
  email: [e-mail]
  scope:
  - https://www.googleapis.com/auth/devstorage.read_write
  - https://www.googleapis.com/auth/logging.write
  - https://www.googleapis.com/auth/service.management
  - https://www.googleapis.com/auth/servicecontrol
  - https://www.googleapis.com/auth/cloud-platform
  - https://www.googleapis.com/auth/pubsub
shieldedInstanceConfig: {}
state: READY

Supprimer la VM TPU

Lorsque vous n'avez plus besoin de la VM TPU, veillez à la supprimer.

gcloud compute tpus tpu-vm delete tpu-name --zone zone