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 dans Superordinateur spécifique à un domaine pour l'entraînement des neurones profonds Réseaux.

Configurations

Un pod TPU v2 est composé de 512 puces interconnectées avec des configurations les liaisons à haut débit. Pour créer une tranche de pod TPU v2, utilisez l'indicateur --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 spécifiant la version de TPU et le nombre de cœurs TPU. Pour 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.17.0-pod-pjrt

Pour en savoir plus sur la gestion des TPU, consultez la page Gérer les TPU. Pour en savoir plus sur le système TPU, d'architecture Cloud TPU, consultez la page Présentation architecture.

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)