Régions et zones des TPU

Présentation

Les principales différences entre les types de TPU sont le prix, les performances, la capacité de mémoire et la disponibilité zonale.

Google Cloud définit les zones géographiques à l'aide de régions subdivisées en zones l'emplacement des ressources de calcul physiques. Par exemple, La région us-central1 est une région proche du centre géographique de aux États-Unis. Lorsque vous créez une VM TPU, vous spécifiez la zone dans laquelle vous que vous voulez le créer. Reportez-vous à la documentation Global, régional et zonal de Compute Engine ressources pour en savoir plus sur les ressources régionales et zonales.

Vous pouvez créer des configurations TPU dans les zones indiquées dans le tableau suivant.

États-Unis

Type de TPU (v2) Cœurs de TPU v2 Zones disponibles
v2-8 8 us-central1-b
us-central1-c
us-central1-f
v2-32 32 us-central1-a
v2-128 128 us-central1-a
v2-256 256 us-central1-a
v2-512 512 us-central1-a
Type de TPU (v3) Cœurs de TPU v3 Zones disponibles
v3-8 8 us-central1-a
us-central1-b
us-central1-f
Type de TPU (v4) Puces TPU v4 Zones disponibles
Toutes les configurations v4 varie selon la taille des tranches us-central2-b
Type de TPU (v5e) Puces TPU v5e Zones disponibles
Toutes les configurations v5litepod varie en fonction de la taille des segments us-central1-a
us-east5-a
us-east5-b
us-east5-c
us-south1-a
us-west1-c
us-west4-a
us-west4-b
Type de TPU (v5p) Puces TPU v5p Zones disponibles
Toutes les configurations v5p varie en fonction de la taille des segments us-east5-a

Europe

Type de TPU (v2) Cœurs de TPU v2 Zones disponibles
v2-8 8 europe-west4-a
v2-32 32 europe-west4-a
v2-128 128 europe-west4-a
v2-256 256 europe-west4-a
v2-512 512 europe-west4-a
Type de TPU (v3) Cœurs de TPU v3 Zones disponibles
v3-8 8 europe-west4-a
v3-32 32 europe-west4-a
v3-64 64 europe-west4-a
v3-128 128 europe-west4-a
v3-256 256 europe-west4-a
v3-512 512 europe-west4-a
v3-1024 1 024 europe-west4-a
v3-2048 2 048 europe-west4-a
Type de TPU (v5e) Puces TPU v5e Zones disponibles
V5lite-1 1 europe-west4-b
v5lite-4 4 europe-west4-b
v5lite-8 8 europe-west4-b
Toutes les configurations v5litepod varie selon la taille des tranches europe-west1-b
europe-west4-a
europe-west4-b
Type de TPU (v5p) Puces TPU v5p Zones disponibles
Toutes les configurations v5p varie selon la taille des tranches europe-west4-b

Asie-Pacifique

Type de TPU (v2) Cœurs de TPU v2 Zones disponibles
v2-8 8 asia-east1-c
Type de TPU (v5e) Puces TPU v5e Zones disponibles
Toutes les configurations v5litepod varie selon la taille des tranches asia-southeast1-b

Les types de TPU dotés d'un grand nombre de puces ou de cœurs ne sont disponibles qu'en quantités limitées. Les types de TPU comprenant un plus faible nombre de puces ou de cœurs sont plus généralement disponibles.

Identifier le meilleur compromis prix/performances

Pour choisir le type de TPU à utiliser, vous pouvez effectuer des tests à l'aide d'un Tutoriel Cloud TPU pour entraîner un modèle semblables à votre application.

Exécutez le tutoriel pour 5 à 10% du nombre d'étapes que vous utiliserez pour effectuer l'ensemble l'entraînement sur un type de TPU v2-8 ou v3-8. Le résultat vous indique le temps nécessaire à l'exécution de ce nombre de pas sur votre modèle pour chaque type de TPU.

Les performances des différents types de TPU évoluent de façon linéaire. Ainsi, si vous connaissez le temps nécessaire à l'exécution d'une tâche sur un type de TPU v2-8 ou v3-8, vous pouvez estimer le temps que cette tâche prendra avec un type de TPU plus grand doté de davantage de puces ou de cœurs.

Par exemple, si un type de TPU v2-8 met 60 minutes pour 10 000 pas, un nœud v2-32 devrait effectuer la même tâche en 15 minutes environ.

Lorsque vous connaissez la durée d'entraînement approximative de votre modèle sur différents types de TPU, vous pouvez comparer le coût VM/TPU par rapport à la durée d'entraînement afin de décider du meilleur compromis.

Pour déterminer la différence de coût entre les différents types de TPU pour Cloud TPU et la VM Compute Engine associée, consultez la page des tarifs de Cloud TPU.

Spécifier le type de TPU

Quel que soit le framework que vous utilisez, vous devez spécifier un v2. ou v3 avec le type de TPU accelerator-type lorsque vous lancer un TPU. Pour un TPU v4 ou version ultérieure, vous pouvez spécifier le type et la taille à l'aide de AcceleratorType ou de AcceleratorConfig. Pour en savoir plus, consultez la section TPU version. Des exemples de commandes sont présentés dans la section Gérer les TPU.

Étape suivante

  • Pour connaître les tarifs des TPU dans chaque région, consultez la page Tarifs.
  • Pour en savoir plus sur l'architecture des TPU, consultez la page Architecture du système.
  • Consultez la section Cas d'utilisation de TPU pour comprendre quels types de modèles sont adaptés à Cloud TPU.