Configura el entorno de Cloud TPU

Antes de que puedas usar los recursos de Cloud TPU para entrenar o ejecutar inferencias en modelos, debes realizar los siguientes 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. registrarte para obtener una cuenta nueva.

  2. Instala Google Cloud CLI. Google Cloud CLI es una 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 el uso de Google Cloud. Solo se te cobrará después de que empiezas 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 del 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 crear un servicio administrado por el usuario cuenta 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 que se indican en Ver acceso actual para ver que tiene acceso a tu proyecto, organización o carpeta. Para ver los tuyos 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 aparece en esa columna, significa que no tienes ningún rol. En la columna Rol de la fila con tu dirección de correo electrónico, comprueba si el de roles incluye los necesarios.

    Si no tienes un rol obligatorio, otorgar el rol o pídele a un administrador que lo haga.

  3. Crea un agente de servicio de Cloud TPU:

    Los agentes de servicio, un tipo de identidad de servicio, Identidades de servicio mantenidas por Google que utiliza el servicio de Cloud TPU para realizar operaciones en tu proyecto de Google Cloud. Por ejemplo, el servicio se usa para crear y administrar los 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 un de servicio otorga acceso a los recursos de Google Cloud. Recomendamos usar administrada por el usuario, otorgando solo los permisos que requiere tu y la VM de TPU. Cuando uses una cuenta de servicio administrada por el usuario, debes especificarla cuando crearás una VM de TPU con la marca --service-account.

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

    2. Sigue las instrucciones que se indican en Administra el acceso a proyectos, carpetas y organizaciones para otorgar tu servicio. de Google Cloud a los servicios de Google Cloud a los que accederá tu TPU. El se recomiendan los siguientes roles para que la TPU acceda servicios de Google Cloud.

Prepárate para crear una Cloud TPU

Antes de crear una Cloud TPU, debes solicitar una cuota. También debes considerar usando recursos en cola, así como los parámetros que quieres usar para configurar tu TPU.

  1. Cuota de solicitudes:

    Si quieres crear una Cloud TPU, tu proyecto de Google Cloud debe tener cuota para la versión y el tamaño de la TPU que quieres crear, y la zona en la que deseas crearla. Por ejemplo, si deseas crear una TPU v4-8 en us-central2-b, haz lo siguiente: solicitarías una cuota de 8 núcleos TPU v4 en us-central2-b. Para ver 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. Diferentes tipos de la cuota tienen diferentes expectativas de disponibilidad. Para obtener más información cómo solicitar una cuota, sus tipos y cómo solicitarla, consulta Cuotas.

  2. Determina si deseas usar recursos en cola.

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

    Existen diferentes comandos de gcloud CLI para trabajar con elementos en cola de Google Cloud. 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 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. Si usas TPU v2 o v3, configura la marca --accelerator-type como vVERSION-TENSORCORES Reemplazar VERSION por el número de versión de TPU que deseas usar. Reemplazar TENSORCORES por la cantidad de TensorCores que deseas usar

      Si quieres personalizar la topología física de tu TPU, usa el Las marcas --version y --topology. Establece la marca --version en la TPU versión que quieres usar. Establece la marca --topology en la topología que quieren usar.

      Para obtener más información sobre las configuraciones de TPU, incluidas las funciones configuraciones y variantes de topología, consulta las versiones de TPU.

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

    • Cuenta de servicio: Establece --service-account en la dirección de correo electrónico de un servicio. de servicio, si creaste una, para conectar la cuenta de servicio a tu TPU. Si está vacía, la cuenta de servicio predeterminada de Compute Engine que se usará.

    • Quota type (Tipo de cuota): Agrega la marca que corresponda al tipo de cuota que deseas. usar:

      Tipo de cuota Flag
      Reservado --reserved
      VMs Spot --spot
      Interrumpible mediante recursos en cola --best-effort
      Interrumpible que no usa recursos en cola --preemptible
      A pedido No se necesita 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 siguientes secciones en Administra TPU para obtener más información:

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

¿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.