TPU v2

במאמר הזה מתוארת הארכיטקטורה של Cloud TPU v2 וההגדרות הנתמכות שלו.

ארכיטקטורת המערכת

פרטים על הארכיטקטורה ומאפייני הביצועים של TPU v2 זמינים במאמר A Domain Specific Supercomputer for Training Deep Neural Networks.

הגדרות אישיות

פרוסת TPU v2 מורכבת מ-512 צ'יפים שמחוברים ביניהם באמצעות קישורים מהירים שניתנים להגדרה מחדש. כדי ליצור פרוסת TPU v2, משתמשים בדגל --accelerator-type בפקודת יצירת ה-TPU ‏ (gcloud compute tpus tpu-vm). מציינים את סוג המאיץ על ידי ציון גרסת ה-TPU ומספר ליבות ה-TPU. לדוגמה, כדי להשתמש ב-TPU יחיד מדור שני, משתמשים ב---accelerator-type=v2-8. כדי להשתמש בפרוסת v2 עם 128 TensorCores, משתמשים ב---accelerator-type=v2-128.

הפקודה הבאה מראה איך ליצור פרוסת TPU מדור v2 עם 128 ליבות Tensor:

  $ gcloud compute tpus tpu-vm create tpu-name \
    --zone=us-central1-a \
    --accelerator-type=v2-128 \
    --version=tpu-ubuntu2204-base

מידע נוסף על ניהול TPU זמין במאמר ניהול TPU. מידע נוסף על ארכיטקטורת המערכת של Cloud TPU זמין במאמר ארכיטקטורת המערכת.

בטבלה הבאה מפורטים סוגי ה-TPU v2 הנתמכים:

גרסת TPU סיום התמיכה
v2-8 (תאריך הסיום עדיין לא הוגדר)
v2-32 (תאריך הסיום עדיין לא הוגדר)
v2-128 (תאריך הסיום עדיין לא הוגדר)
v2-256 (תאריך הסיום עדיין לא הוגדר)
v2-512 (תאריך הסיום עדיין לא הוגדר)