TPU v2

Este documento describe la arquitectura y las configuraciones admitidas de Cloud TPU v2

Arquitectura del sistema

Los detalles arquitectónicos y las características de rendimiento de TPU v2 están disponibles en Un supercomputadora específico de dominio para entrenar redes neuronales profundas Redes.

Configuraciones

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

En el siguiente comando, se muestra cómo crear una porción de pod de TPU v2 con 128 núcleos de Tensor:

  $ gcloud compute tpus tpu-vm create tpu-name \
    --zone=zone \
    --accelerator-type=v2-128 \
    --version=tpu-vm-tf-2.17.0-pod-pjrt

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 TPU 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)