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