OS イメージと Docker イメージ

Google Cloud は、一般的なオペレーティング システム、フレームワーク、ライブラリ、ドライバを含むイメージを提供します。 Google Cloud は、これらの事前構成されたイメージを最適化して、AI と ML のワークロードをサポートします。

このドキュメントでは、AI Hypercomputer 環境でワークロードをデプロイ、管理、実行するために使用するイメージの概要について説明します。

画像カテゴリについて

画像は次のカテゴリに分類されます。

  • AI および ML のフレームワークとライブラリ: ML モデルの作成、トレーニング、使用を簡素化するために、ML フレームワークとライブラリのバイナリが事前に構成されている Docker イメージ。AI Hypercomputer では、ディープ ラーニング ソフトウェア レイヤ（DLSL）の Docker イメージを使用して、Google Kubernetes Engine（GKE）クラスタで NeMO や MaxText などの ML モデルを実行できます。
  • クラスタのデプロイとオーケストレーション: AI ワークロードを実行するパフォーマンス最適化インフラストラクチャのデプロイと管理に使用するオペレーティング システム（OS）イメージ。使用するイメージは、使用している GPU インフラストラクチャ（クラスタ化された GPU または一般的な GPU）と、クラスタのデプロイ方法によって異なります。
    • GKE ノードイメージ: これらのイメージを使用して GKE クラスタをデプロイできます。
    • Slurm OS イメージ: Cluster Toolkit は、Slurm ノードに必要なシステム ソフトウェアをインストールするイメージをビルドしてデプロイします。
    • アクセラレータ OS イメージ: これらのイメージを使用して、Compute Engine インスタンスを個別に、またはグループで作成できます。

AI と ML のフレームワークとライブラリ

Google Cloud には、一般的な AI および ML フレームワークとライブラリをパッケージ化した Docker イメージが用意されています。これらのイメージは、AI Hypercomputer で実行されている AI 最適化クラスタでのモデルの開発、トレーニング、デプロイを簡素化するために必要なソフトウェアを提供します。

JAX AI 画像

Google Cloud TPU と GPU 用の JAX AI イメージ（JAII、旧称 JAX Stable Stack イメージ）は、JAX フレームワーク、互換性のあるライブラリのキュレートされたコレクション、 Google Cloud インフラストラクチャの設定を含む、すぐに使用できる Docker イメージを提供します。JAX AI TPU イメージには、JAX ライブラリと TPU ライブラリが事前に構成されています。JAX AI GPU イメージには、JAX ライブラリと関連する CUDA/NVIDIA ライブラリが事前に構成されています。

JAX AI イメージの図

ハードウェア レイヤ

JAX AI イメージは、アクセラレータ（TPU または GPU）とそれに関連付けられた VM で構成されるハードウェア レイヤの上に配置されます。JAX AI イメージを使用するには、TPU または GPU VM をプロビジョニングする必要があります。これを行うには、TPU API、Compute Engine API、または GKE API を使用します。

フレームワーク レイヤ

フレームワーク レイヤは、ML ワークロードを構築するためのツールとライブラリを提供します。JAX AI イメージは、コア JAX ライブラリやその他の重要な依存関係など、JAX ベースの ML ワークロード用に事前構成されたベースを提供し、一貫性のある高パフォーマンスの開発エクスペリエンスを実現します。

JAX AI イメージの LibTPU レイヤは、それぞれの JAX バージョン用に特別にビルドされ、バンドルされています。別の JAX バージョンを使用すると、予期しない動作やエラーが発生する可能性があります。

JAX AI イメージの CUDA レイヤには、GPU トレーニング イメージのベースイメージとして使用される NGC CUDA ディープ ラーニング イメージなど、NVIDIA が管理するコンポーネントが含まれています。GPU イメージには、NVIDIA GPU で Transformer モデルを高速化するためのカスタム NVIDIA ライブラリである Transformer Engine も含まれています。

特定の ML ワークロードでは、JAX AI イメージで提供されるパッケージ以外に、アプリケーション固有の追加パッケージが必要になる場合があります。

JAX AI イメージのライブラリ:

TPU イメージ

機能 パッケージ名
コア ライブラリまたはコンポーネント
ML フレームワーク JAX と JAX lib
LibTPU Cloud LibTPU 安定版リリース
レイヤ / モデル ライブラリ Flax
チェックポイント処理 Orbax
量子化 Qwix
オプティマイザー Optax

aqtp
カスタム カーネル Tokamax
構成 Fiddle
入力パイプライン tf.data

PyGrain array-record

プロファイリング、デバッグ TensorBoard
utils 一般的なループ ユーティリティ

ML グッドプット測定 ML コレクション

クラウド固有のツール
Google Cloud

Google Cloud SDK Google Cloud ストレージ

診断

Cloud Accelerator Diagnostics Cloud TPU Diagnostics

GPU イメージ

機能 パッケージ名
コア ライブラリまたはコンポーネント
ML フレームワーク JAX と JAX lib
NVIDIA CUDA ライブラリ CUDA DL イメージ
レイヤ / モデル ライブラリ Flax
チェックポイント処理 Orbax
オプティマイザー

Optax aqtp TransformerEngine

構成 Fiddle
入力パイプライン

tf.data PyGrain array-record

プロファイリング、デバッグ TensorBoard
utils

一般的なループユーティリティ ML グッドプット測定 ML コレクション

クラウド固有のツール
Google Cloud

Google Cloud SDK Google Cloud ストレージ

診断 Cloud Accelerator Diagnostics

現在の JAX AI イメージ

アプリケーション レイヤ

フレームワーク レイヤの上にあるアプリケーション レイヤに、特定の ML ワークロードを実装します。アプリケーション レイヤには、アプリケーション固有のコード、モデル、ロジックが含まれています。これらはすべて、フレームワーク レイヤが提供するツールとライブラリを使用して構築されています。

このイメージは、JAX ベースの AI ワークロードに堅牢で十分にテストされた基盤を提供しますが、アプリケーション固有の依存関係を追加する必要がある場合があります。その際、JAX とそのコア依存関係を含む、事前構成されたベースレイヤとの干渉を最小限に抑える方法で行うことをおすすめします。既存の依存関係をオーバーライドまたは競合するアプリケーション レベルの依存関係を導入すると、次のような副作用が発生する可能性があります。

  • 予期しない動作: ML ワークロードは、JAX AI イメージに追加の依存関係を導入する前とは異なる動作を示すことがあります。
  • パフォーマンスの低下: 最適化された JAX 関連ライブラリをオーバーライドすると、JAX AI 画像によって提供されるパフォーマンス上のメリットに悪影響を及ぼす可能性があります。
  • 安定性の問題: 追加した依存関係とコア JAX 依存関係の競合により、アプリケーション内で不安定性やランタイム エラーが発生する可能性があります。

リリースの頻度

当初、JAX AI イメージは四半期ごとに提供されます。近い将来、すべての JAX リリースと同期したリリース スケジュールが実現される予定です。これにより、最新の機能や改善が利用可能になるとすぐに利用できるようになります。

サポート

各 JAX AI イメージのリリースは、期間限定のサポート ライフサイクルに準拠しています。この期間内に、既存の JAX AI イメージの変更リクエストの特定のカテゴリに対応します。

  • セキュリティの脆弱性: JAX Stable Stack Docker イメージのベースイメージまたは依存関係で検出されたセキュリティの脆弱性への対処を優先します。潜在的なリスクを軽減するために、更新されたイメージがリリースされます。
  • 互換性を破る変更: JAX AI 画像で使用されている基盤となるライブラリまたはフレームワークに互換性を破る重大な変更があった場合、 Google Cloudは互換性を維持するために必要な更新を評価して実装します。これには、更新された依存関係を使用して Docker イメージを再ビルドすることが含まれる場合があります。

JAII 内のライブラリでセキュリティの脆弱性やバグが発見された場合、更新されたライブラリを JAII に組み込み、他のすべてのライブラリ バージョンを固定して、全体的な安定性を維持します。これにより、新しい JAII リビジョンが作成されます。

リビジョンの変更を最小限に抑える:

JAX-0.4.30-rev1 内のパッケージ「X」にバグが見つかった場合、他のすべてのパッケージを変更せずに、「X」を次のリリース（v2.0 など）に更新します。これにより、新しいリビジョン JAX-0.4.30-rev2 が作成され、できるだけ早くリリースされます。

ディープ ラーニング ソフトウェア レイヤ（DLSL）の Docker イメージ

これらのイメージには、NVIDIA CUDA、NCCL、ML フレームワーク、モデルがパッケージ化されています。ディープ ラーニング ワークロードですぐに使用できる環境を提供します。これらのビルド済みの DLSL Docker イメージは、内部の再現性と回帰テスト中にテストおよび検証されるため、GKE クラスタとシームレスに連携します。

DLSL Docker イメージには、次の利点があります。

  • 事前構成済みのソフトウェア: DLSL Docker イメージは、内部の再現性と回帰テストで使用される設定を複製します。これらのイメージは、事前構成、テスト、最適化された環境を提供するため、インストールと構成にかかる時間と労力を大幅に削減できます。
  • バージョン管理: DLSL Docker イメージは頻繁に更新されます。これらのバージョン アップデートにより、フレームワークとドライバの最新の安定版が提供されます。また、アップデートではセキュリティ パッチも適用されます。
  • インフラストラクチャの互換性: DLSL Docker イメージは、AI Hypercomputer で使用可能な GPU マシンタイプとシームレスに連携するようにビルドされ、テストされています。
  • クイックスタートの手順: 一部の DLSL Docker イメージには、事前構成済みのイメージを使用するワークロードを開始する方法を示すサンプル レシピが付属しています。

NeMo + PyTorch + NCCL gIB プラグイン

これらの Docker イメージは、NVIDIA NeMo NGC イメージに基づいています。これらには Google の NCCL gIB プラグインが含まれており、サポートされている各アクセラレータ マシンでワークロードを実行するために必要なすべての NCCL バイナリがバンドルされています。これらのイメージには、Google Kubernetes Engine にワークロードをデプロイするための gcsfuse や gcloud CLI などの Google Cloudツールも含まれています。

DLSL イメージのバージョン 依存関係のバージョン マシンシリーズ リリース日 サポート終了日 DLSL イメージ名
nemo25.04-gib1.0.6-A4
  • NeMo NGC:25.04.01
  • NCCL giB plugin: 1.0.6
 A4 2025 年 7 月 3 日 2026 年 7 月 3 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo25.04-gib1.0.6-A4
nemo25.04-gib1.0.6-A3U
  • NeMo NGC:25.04.01
  • NCCL giB plugin: 1.0.6
 A3 Ultra 2025 年 7 月 3 日 2026 年 7 月 3 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo25.04-gib1.0.6-A3U
nemo25.02-gib1.0.5-A4
  • NeMo NGC:25.02
  • NCCL giB plugin: 1.0.5
 A4 2025 年 3 月 14 日 2026 年 3 月 14 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo25.02-gib1.0.5-A4
nemo24.07-gib1.0.2-A3U
  • NeMo NGC:24.07
  • NCCL giB plugin: 1.0.2
 A3 Ultra 2025 年 2 月 2 日 2026 年 2 月 2 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo24.07-gib1.0.2-A3U
nemo24.07-gib1.0.3-A3U
  • NeMo NGC:24.07
  • NCCL giB plugin: 1.0.3
 A3 Ultra 2025 年 2 月 2 日 2026 年 2 月 2 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo24.07-gib1.0.3-A3U
nemo24.12-gib1.0.3-A3U
  • NeMo NGC:24.12
  • NCCL giB plugin: 1.0.3
 A3 Ultra 2025 年 2 月 7 日 2026 年 2 月 7 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo24.12-gib1.0.3-A3U
nemo24.07-tcpx1.0.5-A3Mega
  • NeMo NGC:24.07
  • GPUDirect-TCPX: 1.0.5
 A3 Mega 2025 年 3 月 12 日 2026 年 3 月 12 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo24.07-tcpx1.0.5-A3Mega
nemo24.07-tcpx1.0.5-A3High
  • NeMo NGC:24.07
  • GPUDirect-TCPX: 1.0.5
 A3 High（8 個の GPU） 2025 年 3 月 12 日 2026 年 3 月 12 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo-nccl:nemo24.07-tcpx1.0.5-A3High

NeMo + PyTorch

この Docker イメージは NVIDIA NeMo NGC イメージに基づいており、Google Kubernetes Engine にワークロードをデプロイするための gcsfuse や gcloud CLI などの Google Cloud ツールが含まれています。

DLSL イメージのバージョン 依存関係のバージョン マシンシリーズ リリース日 サポート終了日 DLSL イメージ名
nemo24.07--A3U NeMo NGC:24.07 A3 Ultra 2024 年 12 月 19 日 2025 年 12 月 19 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo:nemo24.07-A3U
nemo24.07-tcpx1.0.5-A3Mega
  • NeMo NGC:24.07
  • GPUDirect-TCPX: 1.0.5
 A3 Mega 2025 年 3 月 12 日 2026 年 3 月 12 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo:nemo24.07-tcpx1.0.5-A3Mega
nemo24.07-tcpx1.0.5-A3High
  • NeMo NGC:24.07
  • GPUDirect-TCPX: 1.0.5
 A3 High（8 個の GPU） 2025 年 3 月 12 日 2026 年 3 月 12 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/pytorch-gpu-nemo:nemo24.07-tcpx1.0.5-A3High

MaxText + JAX ツールボックス

この Docker イメージは NVIDIA JAX ツールボックス イメージに基づいており、Google Kubernetes Engine にワークロードをデプロイするための gcsfuse や gcloud CLI などのGoogle Cloud ツールが含まれています。

DLSL イメージのバージョン 依存関係のバージョン マシンシリーズ リリース日 サポート終了日 DLSL イメージ名
toolbox-maxtext-2025-01-10-A3U JAX toolbox: maxtext-2025-01-10 A3 Ultra 2025 年 3 月 11 日 2026 年 3 月 11 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/jax-maxtext-gpu:toolbox-maxtext-2025-01-10-A3U
jax0.5.1-cu12-A3Mega
  • JAX: 0.5.1
  • CUDA: 12.x
  • MaxText: Latest
 A3 Mega 2025 年 3 月 17 日 2026 年 3 月 17 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/jax-gpu-maxtext:jax0.5.1-cu12-A3Mega
jax0.5.1-cu12-A3High
  • JAX: 0.5.1
  • CUDA: 12.x
  • MaxText: Latest
 A3 High（8 個の GPU） 2025 年 3 月 17 日 2026 年 3 月 17 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/jax-gpu-maxtext:jax0.5.1-cu12-A3High

MaxText + JAX Stable Stack

この Docker イメージは、JAX Stable Stack と MaxText に基づいています。このイメージには、Google Kubernetes Engine でワークロードを実行するための dnsutils などの依存関係も含まれています。

DLSL イメージのバージョン 依存関係のバージョン マシンシリーズ リリース日 サポート終了日 DLSL イメージ名
jax-maxtext-gpu:jax0.5.1-cuda_dl25.02-rev1-maxtext-20150317
  • JAX Stable stacks:jax0.5.1-cuda_dl25.02-rev1
  • maxtext commit: 54e98c9e62caa426cf5902be068533ddb4fb79f5
 A4 2025 年 3 月 17 日 2026 年 3 月 17 日 us-central1-docker.pkg.dev/deeplearning-images/reproducibility/jax-maxtext-gpu:jax0.5.1-cuda_dl25.02-rev1-maxtext-20150317

クラスタのデプロイとオーケストレーション

OS イメージには、コンピューティング インスタンスまたは GKE ノードにオペレーティング システムをデプロイするために必要なすべてのソフトウェア コンポーネントが含まれています。オペレーティング システムは、アクセラレータやネットワーキングなどの基盤となるハードウェア リソースを管理します。これにより、AI ワークロードのコンピューティング リソースが提供されます。

GKE ノードイメージ

GKE は、ノードイメージを使用してクラスタをデプロイします。これらのノードイメージは、Container-Optimized OS、Ubuntu、Windows Server などのさまざまなオペレーティング システムで使用できます。GKE Autopilot クラスタのデプロイに必要な containerd（cos_containerd）を含む Container-Optimized OS ノードイメージには、AI / ML ワークロードをサポートするための最適化が含まれています。

これらのノードイメージの詳細については、ノードイメージをご覧ください。

Slurm OS イメージ

Slurm クラスタは、コンピューティング ノードとコントローラ ノードを Compute Engine のコンピューティング インスタンスとしてデプロイします。

AI 最適化 Slurm クラスタをプロビジョニングするには、Cluster Toolkit を使用する必要があります。デフォルトでは、A4X、A4、A3 Ultra クラスタのクラスタ ブループリントは、ビルド済みの Advanced Compute Images を使用するコンピューティング インスタンスをデプロイします。他のマシンシリーズの場合、または Slurm カスタム イメージをビルドする場合、クラスタ ブループリントは、デプロイ中にカスタム OS イメージを自動的にビルドします。このイメージは、クラスタとワークロード マネージャーの管理に必要なシステム ソフトウェアを Slurm ノードにインストールします。デフォルトのブループリントをデプロイする前に変更して、イメージに含まれるソフトウェアをカスタマイズできます。

次のセクションでは、クラスタ ブループリントが A4、A3 Ultra、A3 Mega、A3 High（8 GPU）Slurm ノードにインストールするソフトウェアについて説明します。クラスタ ブループリントは、Ubuntu LTS アクセラレータ OS イメージを拡張します。

A4X Max

GitHub で入手できる A4X Max ブループリントには、デフォルトで次のソフトウェアが含まれています。

  • Ubuntu 24.04 LTS
  • Slurm: バージョン 25.05.2
  • 次の Slurm 依存関係:
    • munge
    • mariadb
    • libjwt
    • lmod
  • Open MPI: バージョン 5.0.x
  • NFS クライアントとサーバー
  • NVIDIA 580 シリーズ ドライバ
  • NVIDIA Enroot
  • NVIDIA Pyxis
  • 次の NVIDIA ツール:
    • Data Center GPU Manager（dcgmi）
    • nvidia-container-toolkit: バージョン 1.17.7
    • nvidia-imex
    • nvidia-fabricmanager
  • CUDA ツールキット: バージョン 13.0
  • A4X Max 用 NCCL プラグイン（nccl-gib-a4x-max-arm64）
  • DOCA-OFED ドライバ
  • Ops エージェント
  • Cloud Storage FUSE

A4X

GitHub で入手できる A4X ブループリントでは、デフォルトでビルド済みの Advanced Compute Images が使用されます。aci-gpu-u2404-slurm-2511-cuda-130-nvidia-580-arm64 イメージ ファミリーには、デフォルトで次のソフトウェアと構成が含まれています。

  • Ubuntu 24.04 LTS（Arm64）
  • Slurm: バージョン 25.11 とすべての依存パッケージ
  • NVIDIA 580 シリーズ ドライバ
  • CUDA ツールキット: バージョン 13.0
  • NVIDIA Container Toolkit、Enroot、Pyxis
  • データセンター GPU Manager（dcgmi）と Fabric Manager
  • NCCL/gIB プラグイン
  • Ops エージェント
  • Cloud Storage FUSE

プリインストールされているツールとマニフェストの詳細の一覧については、Advanced Compute Images の概要をご覧ください。

A4

GitHub で入手できる A4 ブループリントでは、デフォルトで事前ビルドされた Advanced Compute Images が使用されます。aci-gpu-u2404-slurm-2511-cuda-130-nvidia-580-amd64 イメージ ファミリーには、デフォルトで次のソフトウェアと構成が含まれています。

  • Ubuntu 24.04 LTS（x86_64）
  • Slurm: バージョン 25.11 とすべての依存パッケージ
  • NVIDIA 580 シリーズ ドライバ
  • CUDA ツールキット: バージョン 13.0
  • NVIDIA Container Toolkit、Enroot、Pyxis
  • データセンター GPU Manager（dcgmi）と Fabric Manager
  • NCCL/gIB プラグイン
  • InfiniBand とマルチキュー ネットワーキングのサポート
  • Ops エージェント
  • Cloud Storage FUSE

プリインストールされているツールとマニフェストの詳細の一覧については、Advanced Compute Images の概要をご覧ください。

A3 Ultra

GitHub で入手できる A3 Ultra ブループリントでは、デフォルトでビルド済みの Advanced Compute Images が使用されます。aci-gpu-u2404-slurm-2511-cuda-130-nvidia-580-amd64 イメージ ファミリーには、デフォルトで次のソフトウェアと構成が含まれています。

  • Ubuntu 24.04 LTS（x86_64）
  • Slurm: バージョン 25.11 とすべての依存パッケージ
  • NVIDIA 580 シリーズ ドライバ
  • CUDA ツールキット: バージョン 13.0
  • NVIDIA Container Toolkit、Enroot、Pyxis
  • データセンター GPU Manager（dcgmi）と Fabric Manager
  • NCCL/gIB プラグイン
  • InfiniBand とマルチキュー ネットワーキングのサポート
  • Ops エージェント
  • Cloud Storage FUSE

プリインストールされているツールとマニフェストの詳細の一覧については、Advanced Compute Images の概要をご覧ください。

A3 Mega

GitHub で入手できる A3 Mega ブループリントには、デフォルトで次のソフトウェアが含まれています。

  • Ubuntu 22.04 LTS
  • Slurm: バージョン 24.11.2
  • 次の Slurm 依存関係:
    • munge
    • mariadb
    • libjwt
    • lmod
  • Open MPI: 4.1.x の最新リリース
  • PMIx: バージョン 4.2.9
  • NFS クライアントとサーバー
  • NVIDIA 550 シリーズ ドライバ
  • NVIDIA enroot コンテナ ランタイム: バージョン 3.4.1
  • NVIDIA pyxis
  • 次の NVIDIA ツール:
    • Data Center GPU Manager（dcgmi）
    • libnvidia-cfg1-550-server
    • libnvidia-nscq-550
    • nvidia-compute-utils-550-server
    • nsight-compute
    • nsight-systems
  • CUDA ツールキット: バージョン 12.4
  • ibverbs-utils などの Infiniband のサポート
  • A3 Mega 用 TCPXO: NCCL TCPXO と 1.6 Tbps ネットワーク スタック用の専用インストーラ
  • Ops エージェント
  • Cloud Storage FUSE

A3 High（8 個の GPU）

GitHub で入手できる A3 High（8 GPU）ブループリントには、デフォルトで次のソフトウェアが含まれています。

  • Ubuntu 22.04 LTS
  • Slurm: バージョン 24.11.2
  • 次の Slurm 依存関係:
    • munge
    • mariadb
    • libjwt
    • lmod
  • Open MPI: 4.1.x の最新リリース
  • PMIx: バージョン 4.2.9
  • NFS クライアントとサーバー
  • NVIDIA 535 または 550 シリーズ ドライバ
  • NVIDIA enroot コンテナ ランタイム: バージョン 3.4.1
  • NVIDIA pyxis
  • 次の NVIDIA ツール:
    • Data Center GPU Manager（dcgmi）
    • libnvidia-cfg1-535/550-server
    • libnvidia-nscq-535/550
    • nvidia-compute-utils-535/550-server
    • nsight-compute
    • nsight-systems
  • CUDA ツールキット: バージョン 12.2 または 12.4
  • ibverbs-utils などの Infiniband のサポート
  • A3 High（8 GPU）用の TCPX: NCCL TCPX（GPUDirect-TCPX）と 800 Gbps ネットワーク スタック（4 NIC）用の専用インストーラ
  • Ops エージェント
  • Cloud Storage FUSE

アクセラレータ OS イメージ

AI Hypercomputer を使用すると、個々のコンピューティング インスタンスまたはコンピューティング インスタンスのグループをプロビジョニングできます。これらのインスタンスを作成する場合は、インスタンスの作成時に OS イメージを指定する必要があります。

Google Cloud には、インスタンス作成用の OS イメージ一式が用意されています。 Google Cloud には、AI 最適化インスタンス用の特殊なアクセラレータ OS イメージ一式も用意されています。これらのアクセラレータ OS イメージには、NVIDIA ドライバ、Mellanox ドライバ、それらの依存関係など、GPU とネットワーキング機能のコアドライバが含まれています。

各 OS の詳細については、Compute Engine ドキュメントのオペレーティング システムの詳細ページをご覧ください。

アクセラレータ OS イメージは、Rocky Linux と Ubuntu LTS オペレーティング システムで使用できます。

Rocky Linux アクセラレータ

各マシンシリーズで使用できる Rocky Linux アクセラレータ OS イメージは次のとおりです。

インフラストラクチャ イメージ ファミリー アーキテクチャ マシンシリーズ イメージ プロジェクト
クラスタ化された GPU rocky-linux-10-optimized-gcp-nvidia-580 x86 A4、A3 Ultra rocky-linux-accelerator-cloud
rocky-linux-10-optimized-gcp-nvidia-580-arm64 Arm A4X rocky-linux-accelerator-cloud
rocky-linux-9-optimized-gcp-nvidia-580 x86 A4、A3 Ultra rocky-linux-accelerator-cloud
rocky-linux-9-optimized-gcp-nvidia-580-arm64 Arm A4X rocky-linux-accelerator-cloud
rocky-linux-8-optimized-gcp-nvidia-580 x86 A4、A3 Ultra rocky-linux-accelerator-cloud
rocky-linux-8-optimized-gcp-nvidia-580-arm64 Arm A4X rocky-linux-accelerator-cloud

Ubuntu LTS アクセラレータ

各マシンシリーズで使用できる Ubuntu LTS アクセラレータ OS イメージは次のとおりです。

インフラストラクチャ イメージ ファミリー アーキテクチャ マシンシリーズ イメージ プロジェクト
クラスタ化された GPU ubuntu-accelerator-2404-arm64-with-nvidia-580 Arm A4X ubuntu-os-accelerator-images
ubuntu-accelerator-2404-amd64-with-nvidia-580 x86 A4、A3 Ultra ubuntu-os-accelerator-images
ubuntu-accelerator-2204-arm64-with-nvidia-580 Arm A4X ubuntu-os-accelerator-images
ubuntu-accelerator-2204-amd64-with-nvidia-580 x86 A4、A3 Ultra ubuntu-os-accelerator-images

次のステップ