Ligue uma TPU a uma rede de VPC partilhada

Importante: este guia explica como configurar TPUs na nuvem que usam uma rede de VPC partilhada gerida centralmente. Este documento pressupõe que tem familiaridade com as VPCs e que criou uma VPC partilhada. Para mais informações sobre como criar e gerir redes VPC partilhada, consulte o artigo Crie e modifique redes VPC partilhada.

Configure um projeto anfitrião de VPC

Tem de conceder à conta de serviço do TPU no seu projeto de serviço autorizações para gerir recursos no projeto anfitrião. Isto é feito através da função "Agente de VPC partilhada da TPU" (roles/tpu.xpnAgent). Execute os seguintes comandos gcloud para conceder esta associação de funções.

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

Nota: pode encontrar o número do projeto de serviço na secção de informações do projeto do painel de controlo da Google Cloud consola.

Crie uma VM da TPU ligada a uma rede de VPC partilhada

Primeiro, determine que tipos e versões de aceleradores estão disponíveis na zona

gcloud compute tpus accelerator-types list --zone zone

gcloud compute tpus versions list --zone zone

Associa uma VM da TPU a uma rede de VPC partilhada quando cria a TPU. Especifique a sua VPC partilhada através da etiqueta --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

Nota: a etiqueta --network tem de ser definida com o nome de rede totalmente qualificado. Por exemplo, projects/my-host-project-id/global/networks/my-network.

Pode verificar se a VM de TPU está ligada à sua VPC partilhada através do comando gcloud describe:

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

A resposta inclui a rede à qual a sua VM da TPU está associada:

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

Elimine a VM da TPU

Quando terminar de usar a VM de TPU, certifique-se de que a elimina.

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