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 イメージを使用するをご覧ください。

次のステップ