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 |
6.17 | ubuntu-os-accelerator-images |
ubuntu-accel-2404-amd64-tpu-tpu7x |
| v6e(Trillium)、v5p | Ubuntu 22.04 LTS |
6.8 | 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 を作成する。