TPU v2
Este documento descreve a arquitetura e as configurações suportadas do a Cloud TPU v2.
arquitetura do sistema.
Detalhes de arquitetura e características de desempenho da TPU v2 estão disponíveis em Um supercomputador específico do domínio para treinar Neural Profundo Redes.
Configurações
Um pod de TPU v2 é composto por 512 chips interconectados com camadas reconfiguráveis
links de alta velocidade. Para criar uma fração do Pod da TPU v2, use o --accelerator-type
no comando de criação de TPU (gcloud compute tpus tpu-vm
). Você especifica
tipo de acelerador especificando a versão da TPU e o número de núcleos de TPU. Para
Por exemplo, para uma única TPU v2, use --accelerator-type=v2-8
. Para uma fração do pod v2
com 128 TensorCores, use --accelerator-type=v2-128
.
O comando a seguir mostra como criar uma fração do Pod de TPU v2 com 128 TensorCores:
$ gcloud compute tpus tpu-vm create tpu-name \
--zone=zone \
--accelerator-type=v2-128 \
--version=tpu-vm-tf-2.17.0-pjrt
Para mais informações sobre como gerenciar TPUs, consulte Gerenciar TPUs. Para mais informações sobre o sistema de TPU, arquitetura do Cloud TPU, consulte Sistemas do Terraform.
A tabela a seguir lista os tipos de TPU v2 compatíveis:
Versão da TPU | Fim da compatibilidade |
---|---|
v2-8 | (data final ainda não definida) |
v2-32 | (data final ainda não definida) |
v2-128 | (data final ainda não definida) |
v2-256 | (data final ainda não definida) |
v2-512 | (data final ainda não definida) |