TPU v2

Ce document décrit l'architecture et les configurations compatibles de Cloud TPU v2.

Architecture du système

Les détails de l'architecture et les caractéristiques de performances du TPU v2 sont disponibles sur la page Superordinateur spécifique à un domaine pour l'entraînement de réseaux de neurones profonds.

Configurations

Un pod TPU v2 est composé de 512 puces interconnectées avec des liaisons à haut débit reconfigurables. Pour créer une tranche de pod TPU v2, utilisez l'option --accelerator-type dans la commande de création de TPU (gcloud compute tpus tpu-vm). Vous spécifiez le type d'accélérateur en indiquant la version de TPU et le nombre de cœurs de TPU. Par exemple, pour un seul TPU v2, utilisez --accelerator-type=v2-8. Pour une tranche de pod v2 avec 128 TensorCores, utilisez --accelerator-type=v2-128.

La commande suivante montre comment créer une tranche de pod TPU v2 avec 128 TensorCores:

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

Pour en savoir plus sur la gestion des TPU, consultez la page Gérer les TPU. Pour en savoir plus sur l'architecture du système TPU dans Cloud TPU, consultez la page Architecture du système.

Le tableau suivant répertorie les types de TPU v2 compatibles:

Version du TPU	Fin de la compatibilité
v2-8	(Date de fin pas encore fixée)
v2-32	(Date de fin pas encore fixée)
v2-128	(Date de fin pas encore fixée)
v2-256	(Date de fin pas encore fixée)
v2-512	(Date de fin pas encore fixée)