TPU mit einem gemeinsam genutzten VPC-Netzwerk verbinden

VPC-Hostprojekt konfigurieren

Sie müssen dem TPU-Dienstkonto in Ihrem Dienstprojekt Berechtigungen zum Verwalten von Ressourcen im Hostprojekt erteilen. Dazu verwenden Sie die Rolle „Agent für gemeinsam genutzte VPC für TPUs“ (roles/tpu.xpnAgent). Führen Sie die folgenden gcloud-Befehle aus, um diese Rollenbindung zu gewähren.

gcloud projects add-iam-policy-binding host-project-id \
--member=serviceAccount:service-your-service-project-number@gcp-sa-tpu.iam.gserviceaccount.com \
--role=roles/tpu.xpnAgent

TPU-VM erstellen, die mit einem gemeinsam genutzten VPC-Netzwerk verbunden ist

Zuerst müssen Sie ermitteln, welche Beschleunigertypen und ‑versionen in der Zone verfügbar sind.

gcloud compute tpus accelerator-types list --zone zone

gcloud compute tpus versions list --zone zone

Sie verbinden eine TPU-VM dann mit einem gemeinsam genutzten VPC-Netzwerk, wenn Sie die TPU erstellen. Geben Sie Ihre gemeinsam genutzte VPC mit dem Tag --network an:

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

Sie können mit dem Befehl gcloud describe prüfen, ob Ihre TPU-VM mit Ihrer gemeinsam genutzten VPC verbunden ist:

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

Die Antwort enthält das Netzwerk, an das die TPU-VM angehängt ist:

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: 00000000000-compute@developer.gserviceaccount.com
  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

TPU-VM löschen

Wenn Sie mit der TPU-VM fertig sind, löschen Sie sie.

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