Configura el entorno de Cloud TPU

Antes de poder usar los recursos de Cloud TPU para entrenar o ejecutar inferencias en modelos, debes seguir estos pasos de configuración:

Configura el proyecto de Google Cloud

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

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

  2. Instala Google Cloud CLI. Google Cloud CLI es una interfaz para acceder a los recursos y servicios de Google Cloud y administrarlos.

  3. Selecciona o crea un proyecto de Google Cloud:

    En la consola de Google Cloud, selecciona o crea un proyecto de Cloud en el selector de proyectos.

    En Cloud Shell, configura el ID del proyecto con gcloud CLI. El ID del proyecto es el nombre de tu proyecto que se muestra en la consola de Google Cloud.

    $ gcloud config set project PROJECT-ID

  4. Asegúrate de que la facturación esté habilitada para tu proyecto de Google Cloud.

    La configuración de facturación es obligatoria para todo uso de Google Cloud. Solo se te cobrará después de que comiences a usar los recursos de Google Cloud. Para obtener más información, consulta la documentación de 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 muestran 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 una Cloud TPU. También se recomienda que crees una cuenta de servicio administrada por el usuario para conectarla a tu 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

    Console

    1. En la consola de Google Cloud, ve 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 siguientes roles en tu proyecto:

    Sigue las instrucciones en Consulta el acceso actual para visualizar quién tiene acceso a tu proyecto, organización o carpeta. 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 de correo electrónico no está en esa columna, entonces no tienes ningún rol. En la columna Función de la fila con tu dirección de correo electrónico, verifica si la lista de funciones incluye las funciones necesarias.

    Si no tienes una función requerida, otorga la función o pídele a un administrador que lo haga.

  3. Crea una cuenta de servicio de TPU:

    Las cuentas de servicio permiten que el servicio de Cloud TPU acceda a otros servicios de Google Cloud. Usar una cuenta de servicio administrada por el usuario es una práctica recomendada de Google Cloud. Adjunta una cuenta de servicio a tu TPU cuando la crees con la marca --service-account.

    1. Sigue las instrucciones que se indican en Crea cuentas de servicio para crear una cuenta de servicio.

    2. Sigue las instrucciones que se indican 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 las siguientes funciones para que tu TPU acceda a servicios de Google Cloud de uso general.

Prepárate para crear una Cloud TPU

Antes de crear una Cloud TPU, debes solicitar una cuota. También debes considerar el uso de recursos en cola y los parámetros que deseas utilizar para configurar tu TPU.

  1. Cuota de solicitudes:

    A fin de crear una Cloud TPU, tu proyecto de Google Cloud debe tener cuota disponible para la versión y el tamaño de la TPU que quieres crear, así como la zona en la que quieres hacerlo. Por ejemplo, si deseas crear una TPU v4-8 en us-central2-b, deberás solicitar una cuota de 8 núcleos de TPU v4 en us-central2-b. Para obtener más información sobre las zonas donde Cloud TPU está disponible, consulta Regiones y zonas de TPU.

    La cuota se asigna de manera diferente según la versión de TPU. Cada tipo de cuota tiene expectativas de disponibilidad diferentes. Para obtener más información sobre la asignación de cuota, los tipos de cuota y cómo solicitarla, consulta Cuotas.

  2. Determina si quieres usar recursos en cola.

    Crear una Cloud TPU como un recurso en cola es una práctica recomendada. Los recursos en cola te permiten recibir capacidad cuando está disponible. Puedes especificar una hora de inicio y finalización opcional para completar la solicitud.

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

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

    • Zona: Establece la marca --zone en la zona en la que deseas crear una TPU. Debes tener una cuota asignada en 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 TPU v2 o v3, establece la marca --accelerator-type en vVERSION-TENSORCORES. Reemplaza VERSION por el número de versión de TPU que deseas usar. Reemplaza TENSORCORES por la cantidad de TensorCores que deseas usar.

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

      Para obtener más información sobre los parámetros de configuración de TPU, incluidas las configuraciones 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 en 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 VM de TPU.

    • Cuenta de servicio: Configura --service-account como la dirección de correo electrónico de una cuenta de servicio, si creaste una, para conectarla a tu TPU. Si está vacío, se usará la cuenta de servicio predeterminada de Compute Engine.

    • Tipo de cuota: Si deseas crear una TPU con la cuota reservada, agrega la marca --reserved a la solicitud.

      Si deseas crear una TPU con una cuota interrumpible y solicitar un recurso en cola, agrega la marca --best-effort a la solicitud.

      Si deseas crear una TPU con una cuota interrumpible y no solicitas un recurso en cola, agrega la marca --preemptible a la solicitud.

      Si deseas crear una TPU con la cuota a pedido, no necesitas agregar marcas adicionales.

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

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

¿Qué sigue?

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