Conéctate a una VM de TPU sin una dirección IP pública

Si tu organización tiene una restricción de política de la organización constraints/compute.vmExternalIpAccess, debes crear VM de TPU que no tengan una dirección IP externa. Para conectarte a una VM de TPU sin una dirección IP externa, debes hacer lo siguiente:

  1. Habilita el Acceso privado a Google en la subred en la que crearás la VM de TPU.
  2. Otorga roles/iap.tunnelResourceAccessor y roles/tpu.admin a los usuarios que se conectarán a las VM de TPU.
  3. Crea una VM de TPU sin una dirección IP pública.
  4. Establece una conexión SSH a tu VM de TPU con --tunnel-through-iap.

Habilitar el acceso privado a los servicios

Para usar una IAP, debes habilitar el Acceso privado a Google, que te permite conectarte a VM que no tienen direcciones IP externas. En el siguiente comando, reemplaza your-subnet por el nombre de la subred en la que crearás la VM de TPU y your-region por la región en la que se ubicará la VM de TPU.

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

Otorga permisos

Los usuarios que necesitan establecer una conexión SSH a las VM de TPU que no tienen direcciones IP públicas deben recibir la función iap.tunnelResourceAccessor. Para obtener más información sobre cómo otorgar una función, consulta Cómo otorgar una función de IAM.

Crear una VM de TPU sin una dirección IP pública

El siguiente comando muestra cómo crear una VM de TPU sin dirección IP pública.

gcloud alpha compute tpus tpu-vm create \
  --project your-project \
  --zone $ZONE \
  --internal-ips \
  --version tpu-vm-tf-2.6.0 \
  --accelerator-type v2-8 \
  --subnetwork your-subnet \
   $NAME

Establece una conexión SSH a tu VM de TPU con túneles de IAP

El siguiente comando muestra cómo establecer una conexión SSH a una VM de TPU mediante la tunelización de IAP.

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