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.8 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 ディストリビューションまたはバージョンを使用している。
  • カスタム カーネルの変更やセキュリティ構成を適用する。

TPU7x VM には、ゲスト OS の Linux カーネル バージョン 6.8 以降が必要です。

責任の共有モデル

カスタム OS イメージの使用は、次の表に示すように、お客様と Google の間の責任共有モデルに従います。

機能 Google の責任 ユーザーの責任
TPU エージェントとドライバ 変更を公開してエージェントをリリースする エージェントとドライバをインストールして構成する
互換性テスト なし 互換性テストとパフォーマンス テストを実施する
OS のメンテナンス なし OS の更新とライセンスを管理する

独自の Ubuntu イメージのビルドと使用の詳細については、カスタム OS イメージを使用するをご覧ください。

次のステップ