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

Se a sua 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ê criará uma VM da TPU.
  2. Conceda roles/iap.tunnelResourceAccessor e roles/tpu.admin aos usuários que vão se conectar às VMs da TPU.
  3. Criar uma VM da TPU sem um endereço IP público.
  4. Conecte-se via SSH à VM da TPU usando --tunnel-through-iap.

Ativar o acesso a serviços particulares

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

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

Conceder permissões

Os usuários que precisam de SSH em VMs da TPU sem endereços IP públicos precisam receber o papel iap.tunnelResourceAccessor. Para mais informações sobre como conceder um papel, consulte Como 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.17.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 se conectar via SSH a uma VM da TPU usando o tunelamento IAP.

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