TPU ソフトウェア バージョン
このドキュメントでは、Cloud TPU を作成するときに適切な TPU ソフトウェア バージョンを選択する方法について説明します。
TPU リソースの作成時に、ソフトウェア バージョン（ランタイム バージョンとも呼ばれます）を指定します。これは、TPU VM にプリインストールされているソフトウェア環境を指します。これには、Ubuntu オペレーティング システム、Docker、TPU でコードを実行するために必要なその他のソフトウェアが含まれます。
Google Cloud CLI を使用している場合は、--version または --runtime-version パラメータを使用して TPU ソフトウェア バージョンを指定します。 Google Cloud コンソールを使用している場合は、[TPU ソフトウェア バージョン] リストから TPU ソフトウェア バージョンを選択します。
PyTorch と JAX
PyTorch と JAX で一般的な次の TPU ソフトウェア バージョンを使用し、使用するフレームワークをインストールします。
| TPU バージョン | TPU ソフトウェア バージョン |
|---|---|
| Trillium（v6e） | v2-alpha-tpuv6e |
| v5p | v2-alpha-tpuv5 |
| v5e | v2-alpha-tpuv5-lite |
| v4 以前 | tpu-ubuntu2204-base |
PyTorch または JAX のインストールと使用開始の詳細については、PyTorch を使用して Cloud TPU VM で計算を実行すると JAX を使用して Cloud TPU VM で計算を実行するをご覧ください。
TensorFlow
TensorFlow の TPU ソフトウェア バージョンは、特定の命名規則に従います。
tpu-vm-tf-x.y.z[-{pod}][-{device_api}]
x.y.z: TensorFlow のメジャー バージョン、マイナー バージョン、パッチ バージョンを表します。-pod（省略可）: マルチホスト TPU スライスを使用していることを示します。-{device_api}（省略可）: デバイス API を指定します。例: PJRT API を使用している場合は-pjrt。
TPU ソフトウェア バージョンを指定する方法の詳細については、以降のセクションをご覧ください。
TensorFlow のバージョンごとに固有の TPU ソフトウェア バージョンがあります。次の表に、サポートされている 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 |