Choose a TPU v5e slice shapeΒΆ

Choose the slice shape for a TPU v5e TPU VM from the number of chips your workload needs, then check your quota before you create the VM.

Note: A slice shape gives the number of chips along each dimension of the interconnect. A 2D shape is written AxB and a 3D shape AxBxC. Multiply the dimensions to get the number of chips in the slice.

CAUTION:

  • Not enough Cloud TPU quota in the zone.

  • Cloud TPU quota limits the number of TPU cores your project can use in each zone, so a slice larger than your remaining quota can't be created.

  • Check your quota for the zone on the Quotas page of the Google Cloud console, and request an increase before you create the VM.

  1. Find the number of chips your workload needs, and choose the topology that matches it.

    Chips you need

    Topology and hosts

    4

    2x2, half of one host

    8

    2x4, one host

    16

    4x4, two hosts

  2. Check the Cloud TPU quota in the zone where you plan to create the VM.

  3. Create the VM with that topology. For the command, see Create a TPU VM.

Parent topic:TPU v5e

Related information

TPU v5e specifications

TPU v5e supported configurations

TPU v5e VM types