TPU v2

En este documento, se describen la arquitectura y los parámetros de configuración admitidos de Cloud TPU v2.

Arquitectura del sistema

Los detalles de la arquitectura y las características de rendimiento de TPU v2 están disponibles en Una supercomputadora específica de dominio para el entrenamiento de redes neuronales profundas.

Parámetros de configuración

Una porción de TPU v2 se compone de 512 chips interconectados con vínculos de alta velocidad reconfigurables. Para crear una porción de TPU v2, usa la marca --accelerator-type en el comando de creación de TPU (gcloud compute tpus tpu-vm). Especifica el tipo de acelerador indicando la versión de TPU y la cantidad de núcleos de TPU. Por ejemplo, para una sola TPU v2, usa --accelerator-type=v2-8. Para una porción de v2 con 128 TensorCores, usa --accelerator-type=v2-128.

El siguiente comando muestra cómo crear una porción de TPU v2 con 128 TensorCores:

  $ gcloud compute tpus tpu-vm create tpu-name \
    --zone=us-central1-a \
    --accelerator-type=v2-128 \
    --version=tpu-ubuntu2204-base

Para obtener más información sobre la administración de TPU, consulta Administra TPU. Para obtener más información sobre la arquitectura del sistema de Cloud TPU, consulta Arquitectura del sistema.

En la siguiente tabla, se enumeran los tipos de TPU v2 compatibles:

Versión de TPU	Finalización de compatibilidad
v2-8	(Fecha de finalización aún no establecida)
v2-32	(Fecha de finalización aún no establecida)
v2-128	(Fecha de finalización aún no establecida)
v2-256	(Fecha de finalización aún no establecida)
v2-512	(Fecha de finalización aún no establecida)