Comenzar

Antes de comenzar con una de las guías de inicio rápido de la VM de Cloud TPU, lee Introducción a Cloud TPU, que brinda una descripción general del trabajo con las Cloud TPU.

Sigue las instrucciones en Configura el entorno de Cloud TPU y prepárate para crear una Cloud TPU.

Las guías de inicio rápido de VM y Pods proporcionan una breve introducción al trabajo con las VM de Cloud TPU mediante TensorFlow, JAX y PyTorch. En estos temas, se explica cómo instalar un framework de AA y ejecutar una aplicación de ejemplo en una VM de Cloud TPU.

Configura el entorno

• Configura el entorno de Cloud TPU

  Configura tu cuenta y proyecto de Google Cloud para usar las Cloud TPU. Para ello, activa la API de Cloud TPU, instala las herramientas y configura los permisos.

Guías de inicio rápido de VM de Cloud TPU

• Ejecuta TensorFlow en VMs de Cloud TPU

  Ejecuta un cálculo simple en la VM de Cloud TPU con TensorFlow.

• Ejecuta JAX en las VM de Cloud TPU

  Ejecuta un cálculo simple en la VM de Cloud TPU con JAX.

• Ejecuta PyTorch en VMs de Cloud TPU

  Ejecuta un cálculo simple en la VM de Cloud TPU con PyTorch.

Guías de inicio rápido de pod de Cloud TPU

• Entrena en pods de Cloud TPU

  Más información sobre las prácticas recomendadas y los problemas comunes de los pods de Cloud TPU.

• Ejecuta TensorFlow en porciones de pod de Cloud TPU

  Ejecuta un cálculo simple en la porción de pod de Cloud TPU con TensorFlow.

• Ejecuta JAX en porciones de pod de Cloud TPU

  Ejecuta un cálculo simple en la porción de pod de Cloud TPU con JAX.

• Ejecuta PyTorch en porciones de pod de Cloud TPU

  Ejecuta un cálculo simple en la porción de pod de Cloud TPU con PyTorch.