Régions et zones des TPU

Présentation

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

Google Cloud définit l'emplacement géographique des ressources de calcul physiques à l'aide de régions subdivisées en zones. Par exemple, la région us-central1 indique une région proche du centre géographique des États-Unis. Lorsque vous créez une VM TPU, vous spécifiez la zone dans laquelle vous souhaitez la créer. Consultez le document Ressources globales, régionales et zonales de Compute Engine 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 en fonction de la taille des segments 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 en fonction de la taille des segments europe-west1-b
europe-west4-a
europe-west4-b
Type de TPU (v5p) Puces TPU v5p Zones disponibles
Toutes les configurations v5p varie en fonction de la taille des segments 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 en fonction de la taille des segments 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 vous aider à choisir le type de TPU à utiliser, vous pouvez effectuer des tests à l'aide d'un tutoriel Cloud TPU afin d'entraîner un modèle semblable à votre application.

Exécutez le tutoriel pour 5 à 10% du nombre de pas que vous utiliserez pour exécuter l'entraînement complet 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 utilisé, vous spécifiez le type de TPU v2 ou v3 à l'aide du paramètre accelerator-type lorsque vous lancez un TPU. Pour un TPU v4 ou version ultérieure, vous pouvez spécifier le type et la taille à l'aide de AcceleratorType ou AcceleratorConfig. Pour en savoir plus, consultez la section Versions TPU. 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.