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