גרסאות תוכנה של TPU

במאמר הזה מוסבר איך לבחור את גרסת התוכנה המתאימה של TPU כשיוצרים יחידות Cloud TPU.

כשיוצרים משאבי TPU, מציינים את גרסת התוכנה, שנקראת גם גרסת זמן ריצה. גרסה זו מתייחסת לסביבת התוכנה שמותקנת מראש ב-TPU VM. האחסון כולל את מערכת ההפעלה Ubuntu,‏ Docker ותוכנות אחרות שנדרשות להרצת הקוד ב-TPU.

אם אתם משתמשים ב-Google Cloud CLI, אתם מציינים את גרסת התוכנה של ה-TPU באמצעות הפרמטר --version או --runtime-version. אם אתם משתמשים במסוף, אתם בוחרים גרסת תוכנה של TPU מתוך הרשימה TPU software version. Google Cloud

‫PyTorch ו-JAX

משתמשים בגרסאות הנפוצות הבאות של תוכנת TPU ל-PyTorch ול-JAX, ואז מתקינים את ה-framework שרוצים להשתמש בו.

גרסת TPU	גרסת התוכנה של TPU
Trillium (v6e)	v2-alpha-tpuv6e
v5p	v2-alpha-tpuv5
v5e	v2-alpha-tpuv5-lite
גרסה 4 ומטה	tpu-ubuntu2204-base

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

TensorFlow

גרסאות התוכנה של TPU ל-TensorFlow פועלות לפי מוסכמת שמות ספציפית:

tpu-vm-tf-x.y.z[-{pod}][-{device_api}]

• ‫x.y.z: מייצג את הגרסאות הראשית, המשנית ותיקוני הבאגים של TensorFlow.
• ‫-pod (אופציונלי): מציין שאתם משתמשים ב-TPU slice עם כמה מארחים.
• ‫-{device_api} (אופציונלי): מציין את ה-API של המכשיר, לדוגמה, -pjrt (אם משתמשים ב-PJRT API).

בקטעים הבאים מוסבר איך מציינים גרסת תוכנה של TPU.

יש גרסאות תוכנה של TPU שספציפיות לכל גרסה של TensorFlow. בטבלה הבאה מפורטות גרסאות TensorFlow הנתמכות וגרסאות libtpu שמשויכות אליהן:

גרסת TensorFlow	גרסת libtpu.so
2.18.0	1.12.0
2.17.1	1.11.1
2.17.0	1.11.0
2.16.2	1.10.1
2.16.1	1.10.1
2.15.1	1.9.0
2.15.0	1.9.0
2.14.1	1.8.1
2.14.0	1.8.0
2.13.1	1.7.1
2.13.0	1.7.0
2.12.1	1.6.1
2.12.0	1.6.0
2.11.1	1.5.1
2.11.0	1.5.0
2.10.1	1.4.1
2.10.0	1.4.0
2.9.3	1.3.2
2.9.1