Configura el entorno de Cloud TPU

Antes de poder usar los recursos de Cloud TPU para entrenar modelos o ejecutar inferencias en ellos, debes seguir los pasos que se indican a continuación:

Configura tu proyecto de Google Cloud

Debes tener una cuenta y un proyecto de Google Cloud para usar Cloud TPU.

  1. En la consola deGoogle Cloud , accede a tu Cuenta de Google o regístrate para obtener una cuenta nueva.

  2. Crea proyecto de Google Cloud . Para obtener más información, consulta Crea un proyecto.

  3. Habilita la facturación en tu proyecto de Google Cloud .

    Es necesario configurar la facturación para todo el uso de Google Cloud . Solo se te cobrará después de que empieces a usar los recursos de Google Cloud . Para obtener más información, consulta la documentación de la Facturación de Cloud.

    La facturación por el uso de todas las versiones de Cloud TPU sigue los precios regionales estándar que se indican en la página de precios de Cloud TPU.

  4. Instala la Google Cloud CLI. La Google Cloud CLI es un conjunto de bibliotecas y herramientas para acceder a los recursos y servicios de Google Cloud y administrarlos.

  5. Configura el comando gcloud que instalaste como parte de la Google Cloud CLI.

    Ejecuta los comandos siguientes para configurar gcloud de modo que use tu proyecto de Google Cloud.

    $ gcloud config set project PROJECT_ID

  6. Asegúrate de tener habilitada la facturación para tu proyecto de Google Cloud .

    Es necesario configurar la facturación para todo el uso de Google Cloud . Solo se te cobrará después de que empieces a usar los recursos de Google Cloud . Para obtener más información, consulta la documentación de la Facturación de Cloud.

    La facturación por el uso de todas las versiones de Cloud TPU sigue los precios regionales estándar que se indican en la página de precios de Cloud TPU.

Configura tu entorno para usar Cloud TPU

Antes de solicitar una Cloud TPU, debes activar la API de Cloud TPU y asegurarte de tener permisos para administrar el acceso en tu proyecto y crear la Cloud TPU. También te recomendamos crear una cuenta de servicio administrada por el usuario para vincularla a la TPU.

  1. Activa la API de Cloud TPU desde la consola de Google Cloud o con la gcloud CLI en Cloud Shell:

    gcloud

    $ gcloud services enable tpu.googleapis.com

    Consola

    1. En la consola de Google Cloud , accede a la página de la API de Cloud TPU.

    Ir a la página de la API de Cloud TPU

    1. Haz clic en Habilitar.

  2. Asegúrate de tener los roles siguientes en tu proyecto:

    Sigue las instrucciones presentes en Consulta el acceso actual para ver qué principales tienen acceso al proyecto, la carpeta o la organización. Para ver tu propio acceso, en la columna Principal, busca la fila que tiene tu dirección de correo electrónico. Si tu dirección no está en esa columna, significa que no tienes ningún rol. En la columna Rol de la fila en la que figura tu dirección de correo electrónico, verifica si la lista de roles incluye los roles necesarios.

    Si no tienes un rol obligatorio, otórgalo o pídele a un administrador que lo haga.

  3. Crea un agente de servicio de Cloud TPU:

    Los agentes de servicio, son un tipo de identidad de servicio mantenido por Google y usado por el servicio de Cloud TPU para realizar operaciones en tu proyecto de Google Cloud . Por ejemplo, el agente de servicio se usa para crear y administrar recursos de Cloud TPU.

      $ gcloud beta services identity create --service tpu.googleapis.com --project PROJECT_ID

  4. Crea una cuenta de servicio de TPU:

    Las cuentas de servicio actúan como credenciales para tus VMs de TPU. Los roles asociados a una cuenta de servicio otorgan acceso a los recursos de Google Cloud . Te recomendamos que uses una cuenta de servicio administrada por el usuario y que solo otorgues los permisos que requiere tu VM de TPU. Cuando usas una cuenta de servicio administrada por el usuario, debes especificarla cuando creas una VM de TPU con la marca --service-account.

    1. Sigue las instrucciones presentes en Crea cuentas de servicio para crear una de estas cuentas administrada por el usuario.

    2. Sigue las instrucciones presentes en Administra el acceso a proyectos, carpetas y organizaciones para otorgar a tu cuenta de servicio acceso a los servicios de Google Cloud a los que accederá tu TPU. Se recomiendan los roles siguientes para que tu TPU acceda a los servicios deGoogle Cloud que se usan con frecuencia.

Preparación para crear una Cloud TPU

Antes de crear una Cloud TPU, debes solicitar una cuota. También debes considerar usar recursos en cola y qué parámetros quieres usar para configurar tu TPU.

  1. Elige una opción de consumo.

    Las opciones de consumo son los métodos que se usan para solicitar capacidad de Cloud TPU. Elige una opción de consumo basada en los factores que se indican a continuación:

    • La rapidez con la que necesitas la capacidad
    • Cuánto tiempo necesitas la capacidad
    • Si necesitas capacidad para un período fijo o flexible
    • Si tu capacidad se puede interrumpir
    • Precios

    Para obtener más información, consulta Opciones de consumo de Cloud TPU.

  2. Cuota de solicitudes:

    Las cuotas de TPU son límites que controlan la cantidad de núcleos de Cloud TPU que puedes usar con un proyecto deGoogle Cloud . Cada versión de TPU está asociada a su propia cuota. Para crear una Cloud TPU, tu proyecto de Google Cloud debe incluir cuota para la versión y el tamaño de la TPU que quieres crear, además de la zona en la que deseas crearla. Para obtener más información sobre la asignación y los tipos de cuotas, además de cómo solicitarlas, consulta Cuotas.

    Si creas TPU con Google Kubernetes Engine o Vertex AI, necesitarás cuota para esos servicios. Para obtener más información sobre las cuotas de GKE, consulta Planifica la configuración de las TPU con GKE. Para obtener más información sobre las cuotas de Vertex AI, consulta Cuotas y límites de Vertex AI.

  3. Determina si quieres usar recursos en cola.

    Una práctica recomendada consiste en crear una Cloud TPU como un recurso en cola. Los recursos en cola permiten recibir capacidad una vez que esté disponible. Puedes especificar un horario de inicio y de finalización opcionales para cuando se debe completar la solicitud.

    Existen diferentes comandos de la gcloud CLI para trabajar con recursos en cola. Para obtener más información, consulta la guía del usuario sobre los recursos en cola.

  4. Determina los parámetros de creación de la Cloud TPU:

    • Zona: Establece la marca --zone en la zona en la que quieres crear una TPU. Debes tener cuota asignada a esta zona. Para obtener más información, consulta Regiones y zonas de TPU.

    • Configuración de TPU: Si no necesitas especificar una topología personalizada o si usas una TPU v2 o v3, establece la marca --accelerator-type en vVERSION-TENSORCORES. Reemplaza VERSION por el número de versión de la TPU que quieres usar. Reemplaza TENSORCORES por la cantidad de TensorCores que quieres usar.

      Si quieres personalizar la topología física de tu TPU, usa las marcas --version y --topology. Establece la marca --version según la versión de TPU que quieres usar. Establece la marca --topology según la topología que quieres usar.

      Para obtener más información sobre los parámetros de configuración de TPU, incluidos los parámetros compatibles y las variantes de topología, consulta Versiones de TPU.

    • Versión de software: Si solicitas un recurso en cola, establece la marca --runtime-version según el nombre de la versión de software que quieres usar. De lo contrario, usa la marca --version. Las versiones de software de TPU están disponibles para los frameworks de TensorFlow, PyTorch y JAX. Para obtener más información sobre las versiones de software compatibles, consulta Versiones de software de VMs de TPU.

    • Cuenta de servicio: Establece --service-account según la dirección de correo electrónico de una cuenta de servicio (si creaste una) para vincular la cuenta de servicio a tu TPU. Si dejas el campo vacío, se usará la cuenta de servicio predeterminada de Compute Engine.

    • Tipo de cuota: Agrega la marca que corresponde al tipo de cuota que quieres usar:

      Tipo de cuota Marca
      Reservado --reserved
      VMs Spot --spot
      Interrumpible (no disponible para recursos en cola; usa --spot en su lugar) --preemptible
      Bajo demanda No se necesita ninguna marca adicional

    • Configuración avanzada: Puedes agregar marcas adicionales a tu solicitud para configurar la TPU. Consulta la documentación de gcloud compute tpus tpu-vm create y las secciones siguientes para obtener más información:

Para ver ejemplos de cómo crear una Cloud TPU, consulta Comenzar.

¿Qué sigue?

  1. Aprende a crear y administrar recursos de VMs y TPU.
  2. Ejecuta una guía de inicio rápido de Cloud TPU.