Conectar-se a uma VM do TPU sem um endereço IP público

Se a organização tiver uma restrição de política constraints/compute.vmExternalIpAccess, será necessário criar VMs de TPU que não tenham um endereço IP externo. Para se conectar a uma VM da TPU sem um endereço IP externo, você precisa fazer o seguinte:

1. Ative o Acesso privado do Google para a sub-rede em que você vai criar uma VM TPU.
2. Conceda roles/iap.tunnelResourceAccessor e roles/tpu.admin aos usuários que vão se conectar às VMs da TPU.
3. Crie uma VM TPU sem um endereço IP público.
4. Faça SSH na VM da TPU usando --tunnel-through-iap.

Ativar o acesso a serviços particulares

Para usar um IAP, ative o Acesso privado do Google, que permite a conexão com VMs que não têm endereços IP externo. No comando abaixo, substitua your-subnet pelo nome da sub-rede em que você vai criar a VM da TPU e your-region pela região em que a VM da TPU vai estar localizada.

gcloud compute networks subnets update your-subnet \
--region=your-region \
--enable-private-ip-google-access

Conceder permissões

Os usuários que precisam usar SSH em VMs da TPU que não têm endereços IP públicos precisam receber o papel iap.tunnelResourceAccessor. Para mais informações sobre como conceder uma função, consulte Conceder um papel do IAM.

Criar uma VM TPU sem um endereço IP público

O comando a seguir mostra como criar uma VM TPU sem um endereço IP público.

gcloud compute tpus tpu-vm create tpu-vm-name \
  --zone $ZONE \
  --project your-project \
  --internal-ips \
  --version tpu-vm-tf-2.18.0-pjrt \
  --accelerator-type v2-8 \
  --subnetwork your-subnet \

SSH na VM da TPU usando o encapsulamento do IAP

O comando a seguir mostra como fazer SSH em uma VM de TPU usando o túnel IAP.

gcloud alpha compute tpus tpu-vm ssh tpu-vm-name --tunnel-through-iap