カスタム トレーニング用に事前にビルドされたコンテナ

Vertex AI によって、カスタム トレーニング用のビルド済みコンテナとして実行する Docker コンテナ イメージが提供されます。これらのコンテナは、機械学習(ML)フレームワークとフレームワークのバージョンで編成され、トレーニング コードで使用する共通の依存関係が含まれています。多くの場合、トレーニング用の独自のカスタム コンテナを作成するよりも、ビルド済みコンテナを使用するほうが簡単です。

このドキュメントでは、トレーニング用のビルド済みコンテナを一覧表示し、Python トレーニング アプリケーションで使用する方法について説明します。

Vertex AI は、セキュリティの脆弱性を最小限に抑えるため、スケジュールに基づいて各フレームワーク バージョンをサポートします。Vertex AI フレームワークのサポート ポリシーを確認して、サポート終了日と利用可能期間に与える影響を確認してください。

使用可能なコンテナ イメージ

次の各コンテナ イメージは、さまざまなロケーションにデータを格納する複数の Artifact Registry リポジトリで利用できます。カスタム トレーニングを行うときは、任意の画像の URI が使用可能であり、各 URI は同じコンテナ イメージを指定しています。 Google Cloud コンソールを使用してカスタム トレーニングを行う場合、 Google Cloud コンソールは、Vertex AI を使用しているロケーションに最適な URI を選択します。これにより、レイテンシが短縮されます。

TensorFlow

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 対応イメージ
2.17(Python 3.10) CPU のみ 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
2.17(Python 3.10) GPU(CUDA 12.x) 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
2.16(Python 3.10) CPU のみ 2025 年 6 月 28 日 2026 年 6 月 28 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
2.16(Python 3.10) GPU(CUDA 12.x) 2025 年 6 月 28 日 2026 年 6 月 28 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
2.15(Python 3.10) CPU のみ 2024 年 11 月 14 日 2025 年 11 月 14 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
2.15(Python 3.10) GPU(CUDA 12.x) 2024 年 11 月 14 日 2025 年 11 月 14 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
2.15(Python 3.10) TPU のみ 2024 年 10 月 23 日 2025 年 10 月 23 日
  • us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
2.14(Python 3.10) CPU のみ 2024 年 9 月 26 日 2025 年 9 月 26 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
2.14(Python 3.10) GPU(CUDA 11.x) 2024 年 9 月 26 日 2025 年 9 月 26 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
2.13(Python 3.10) CPU のみ 2024 年 7 月 5 日 2025 年 7 月 5 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
2.13(Python 3.10) GPU(CUDA 11.x) 2024 年 7 月 5 日 2025 年 7 月 5 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
2.12(Python 3.10) CPU のみ 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
2.12(Python 3.10) GPU(CUDA 11.x) 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
2.12(Python 3.8) TPU のみ 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
2.11(Python 3.10) CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
2.11(Python 3.10) GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
2.11(Python 3.7) CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
2.11(Python 3.7) GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
2.9(Python 3.10) CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
2.9(Python 3.10) GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
2.9(Python 3.7) CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
2.9(Python 3.7) GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
2.8(Python 3.10) CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
2.8(Python 3.10) GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
2.8(Python 3.7) CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
2.8(Python 3.7) GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest
2.7(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest
2.7(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest
2.6(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest
2.6(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest
2.5(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest
2.5(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest
2.4(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest
2.4(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest
2.3(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest
2.3(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest
2.2(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest
2.2(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest
2.1(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest
2.1(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest
1.15(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest
1.15(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest

PyTorch

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 対応イメージ
2.4(Python 3.10) CPU のみ 2025 年 7 月 24 日 2026 年 7 月 24 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
2.4(Python 3.10) GPU のみ(CUDA 12.x) 2025 年 7 月 24 日 2026 年 7 月 24 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
2.3(Python 3.10) CPU のみ 2025 年 4 月 24 日 2026 年 4 月 24 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest
2.3(Python 3.10) GPU のみ(CUDA 12.x) 2025 年 4 月 24 日 2026 年 4 月 24 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest
2.2(Python 3.10) CPU のみ 2025 年 1 月 30 日 2026 年 10 月 23 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest
2.2(Python 3.10) GPU のみ(CUDA 12.x) 2025 年 1 月 30 日 2026 年 1 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest
2.2(Python 3.10) TPU のみ 2024 年 11 月 6 日 2025 年 11 月 6 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
2.1(Python 3.10) CPU のみ 2024 年 10 月 4 日 2025 年 10 月 4 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest
2.1(Python 3.10) GPU のみ(CUDA 12.x) 2024 年 10 月 4 日 2025 年 10 月 4 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest
2.1(Python 3.8) TPU のみ 2024 年 10 月 23 日 2025 年 10 月 23 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest
2.0(Python 3.10) CPU のみ 2024 年 3 月 15 日 2025 年 3 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest
2.0(Python 3.10) GPU のみ(CUDA 11.x) 2024 年 3 月 15 日 2025 年 3 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest
2.0(Python 3.8) TPU のみ 2024 年 5 月 15 日 2025 年 5 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest
1.13(Python 3.10) CPU または GPU(CUDA 11.x) 2024 年 5 月 15 日 2025 年 5 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest
1.13(Python 3.7) CPU または GPU(CUDA 11.x) 2023 年 12 月 8 日 2024 年 12 月 8 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest
1.12(Python 3.10) CPU または GPU(CUDA 11.x) 2024 年 5 月 15 日 2025 年 5 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest
1.12(Python 3.7) CPU または GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest
1.11(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest
1.11(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest
1.10(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest
1.10(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest
1.9(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest
1.9(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest
1.7(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest
1.7(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest
1.6(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest
1.6(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest
1.4(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest
1.4(Python 3.7) GPU(CUDA 11.x) 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest

scikit-learn

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 対応イメージ
1.6(Python 3.10) CPU のみ 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
1.6(Python 3.10) GPU(CUDA 12.x) 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
1.0(Python 3.10) CPU のみ 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest
1.0(Python 3.10) GPU(CUDA 11.x) 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest
0.23(Python 3.7) CPU のみ 2023 年 9 月 1 日 2024 年 9 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest
  • europe-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest
  • asia-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest

XGBoost

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 サポートされているイメージ
2.1(Python 3.10) CPU のみ 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
2.1(Python 3.10) GPU(CUDA 12.x) 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
1.6(Python 3.10) CPU のみ 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest
1.6(Python 3.10) GPU(CUDA 11.x) 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest
1.1(Python 3.7) CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest

サポートされている Ray on Vertex AI のバージョン

Ray on Vertex AI コンテナ イメージを使用できます。次のバージョンのビルド済みコンテナがサポートされています。

Ray バージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 対応イメージ
2.47.1(Python 3.11) CPU のみ 2026 年 7 月 28 日 2027 年 7 月 28 日
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
2.47.1(Python 3.11) GPU(CUDA 11.x) 2026 年 7 月 28 日 2027 年 7 月 28 日
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
2.42.0(Python 3.11) CPU のみ 2026 年 3 月 1 日 2027 年 3 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
2.42.0(Python 3.11) GPU(CUDA 11.x) 2026 年 3 月 1 日 2027 年 3 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
2.42.0(Python 3.10) CPU のみ 2026 年 3 月 1 日 2027 年 3 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
2.42.0(Python 3.10) GPU(CUDA 11.x) 2026 年 3 月 1 日 2027 年 3 月 1 日
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
2.33.0(Python 3.10) CPU のみ 2025 年 9 月 16 日 2026 年 9 月 16 日
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
2.33.0(Python 3.10) GPU(CUDA 11.x) 2025 年 9 月 16 日 2026 年 9 月 16 日
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
2.9.3(Python 3.10) CPU のみ 2025 年 5 月 7 日 2026 年 5 月 7 日
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
2.9.3(Python 3.10) GPU(CUDA 11.x) 2025 年 5 月 7 日 2026 年 5 月 7 日
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest

ビルド済みコンテナを使用してトレーニングする

ビルド済みコンテナを使用するには、コンテナ設定の構成に関するガイドをご覧ください。

GPU をサポートするコンテナ イメージを使用している場合は、CustomJobHyperparameterTuningJobTrainingPipeline の各リソース作成時に machineSpec.acceleratorTypemachineSpec.acceleratorCount のフィールドを指定してください。

次のステップ