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

Se a organização tiver uma restrição de política da organização constraints/compute.vmExternalIpAccess, você precisará 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, faça 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 se conectarão às VMs da TPU.
  3. Criar uma VM de 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 privado do Google, que permite a conexão com VMs que não têm endereços IP externo. No comando a seguir, substitua your-subnet pelo nome da sub-rede em que você criará a VM da TPU e your-region pela região em que 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 se conectar via SSH às VMs de TPU que não têm 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 de TPU sem um endereço IP público

O comando a seguir mostra como criar uma VM de TPU sem 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.16.1-pjrt \
  --accelerator-type v2-8 \
  --subnetwork your-subnet \

SSH na VM de TPU usando tunelamento IAP

O comando a seguir mostra como se conectar via SSH a uma VM de TPU usando o encapsulamento do IAP.

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