Get started (original) (raw)

Before starting one of the Cloud TPU VM quickstarts, readIntroduction to Cloud TPU which gives an overview of working with Cloud TPUs.

Follow the instructions inSet up the Cloud TPU environment to prepare to create a Cloud TPU.

The VM and slice quickstarts provide a brief introduction to working with Cloud TPU VMs using TensorFlow, JAX, and PyTorch. These topics explain how to install an ML framework and run a sample application on a Cloud TPU VM.

Set up your environment

• Set up the Cloud TPU environment Configure your Google Cloud account and project for using Cloud TPUs, by activating the Cloud TPU API, setting up permissions, and installing tools.

Cloud TPU VM quickstarts

• Run JAX on Cloud TPU VMs Run a simple calculation on Cloud TPU VM with JAX.
• Run PyTorch on Cloud TPU VMs Run a simple calculation on Cloud TPU VM with PyTorch.

Cloud TPU Pod quickstarts

• Train on Cloud TPU Pods Learn about best practices and common issues for Cloud TPU Pods.
• Run JAX on Cloud TPU slices Run a simple calculation on Cloud TPU slice with JAX.
• Run PyTorch on Cloud TPU slices Run a simple calculation on Cloud TPU slice with PyTorch.