TPU OS イメージ
Compute Engine を使用して TPU VM を作成する場合は、Google マネージドの公開オペレーティング システム(OS)イメージまたはカスタム イメージを使用できます。Google マネージド イメージは TPU のパフォーマンス向けに最適化されており、必要なドライバとエージェントが含まれています。デフォルトでは、すべての Google Cloud プロジェクトがこれらの公開 OS イメージにアクセスできます。
Google 管理の OS イメージ
Google は、ワークロードの機能に必要な TPU ランタイム、ドライバ、エージェントが事前構成された Ubuntu ベースの OS イメージを提供しています。
次の表に、サポートされている各 TPU バージョンの Google マネージド OS イメージを示します。
| TPU バージョン | OS バージョン | イメージ プロジェクト | イメージ ファミリー |
|---|---|---|---|
| TPU7x(Ironwood) | Ubuntu 24.04 LTS |
ubuntu-os-accelerator-images |
ubuntu-accel-2404-amd64-tpu-tpu7x |
| TPU v6e(Trillium) | Ubuntu 22.04 LTS |
ubuntu-os-accelerator-images |
ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e |
| TPU v5p | Ubuntu 22.04 LTS |
ubuntu-os-accelerator-images |
ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e |
カスタム OS イメージ
Compute Engine では、TPU VM にカスタム OS イメージを使用できます。カスタム イメージは、次のような場合に便利です。
- カスタム ソフトウェアとライブラリをプリロードして、VM の起動時間を短縮します。
- Google マネージド リストに含まれていない OS ディストリビューションまたはバージョンを使用している。
- カスタム カーネルの変更やセキュリティ構成を適用する。
責任の共有モデル
カスタム OS イメージの使用は、次の表に示すように、お客様と Google の間の責任の共有モデルに従います。
| 機能 | Google の責任 | ユーザーの責任 |
|---|---|---|
| TPU エージェントとドライバ | 変更を公開してエージェントをリリースする | エージェントとドライバをインストールして構成する |
| 互換性テスト | なし | 互換性テストとパフォーマンス テストを実施する |
| OS のメンテナンス | なし | OS の更新とライセンスを管理する |
独自の Ubuntu イメージのビルドと使用の詳細については、カスタム OS イメージを使用するをご覧ください。
次のステップ
- カスタム OS イメージを使用する方法を学習する。
- TPU VM インスタンスを作成します。
- マルチホスト TPU スライスを使用して MIG を作成します。
- 単一ホスト TPU スライスを使用して MIG を作成する。