TPU v2

Este documento descreve a arquitetura e as configurações suportadas do a Cloud TPU v2.

arquitetura do sistema.

Os detalhes de arquitetura e as características de desempenho da TPU v2 estão disponíveis em Um supercomputador específico para domínios para treinamento de redes neurais profundas.

Configurações

Um pod de TPU v2 é composto por 512 chips interconectados com links de alta velocidade reconfiguráveis. 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 da TPU. 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 fatia de 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-pod-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)