Conectar uma TPU a uma rede VPC compartilhada
Configurar um projeto host da VPC
É necessário conceder à conta de serviço da TPU
no seu projeto de serviço
permissões para gerenciar recursos no projeto host.
Para fazer isso, use o papel "Agente de VPC compartilhada para TPU" (roles/tpu.xpnAgent
). Executar
os seguintes comandos gcloud
para conceder essa vinculação de papel.
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
Criar uma VM de TPU conectada a uma rede VPC compartilhada
Primeiro, determine quais 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
Você conecta uma VM da TPU a uma rede VPC compartilhada ao criar a TPU.
Especifique sua VPC compartilhada usando a tag --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
Verifique se a VM da TPU está conectada à VPC compartilhada usando:
o comando gcloud describe
:
$ gcloud compute tpus tpu-vm describe tpu-name --zone zone
A resposta inclui a rede à qual sua VM da TPU está conectada:
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
Excluir a VM da TPU
Quando terminar de usar a VM da TPU, não se esqueça de excluí-la.
gcloud compute tpus tpu-vm delete tpu-name --zone zone