サポートされているフレームワーク

このページでは、次のサービスでサポートされているフレームワークとその構成パッケージの一覧を示します。

フレームワーク サポート ポリシーの詳細については、Agent Platform フレームワーク サポート ポリシーをご覧ください。

カスタム トレーニングでサポートされているフレームワーク

Gemini Enterprise Agent Platform は、コンテナ イメージを使用して、トレーニング リクエストを処理するようにクラウド リソースを構成します。カスタム トレーニングのビルド済みコンテナでは、次のバージョンがサポートされています。

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

予測と説明でサポートされているフレームワーク

Gemini Enterprise Agent Platform は、コンテナ イメージを使用して、予測リクエストを処理するようにクラウド リソースを構成します。予測と説明用のビルド済みコンテナでは、次のバージョンがサポートされています。

TensorFlow

使用可能な TensorFlow コンテナ イメージ(クリックして展開)

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 サポートされているイメージ
2.15 CPU のみ 2026 年 7 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-15:latest
2.15 GPU(CUDA 12.x) 2026 年 7 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-15:latest
2.14 CPU のみ 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-14:latest
2.14 GPU(CUDA 12.x) 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-14:latest
2.13 CPU のみ 2024 年 11 月 28 日 2025 年 11 月 28 日
2.13 GPU(CUDA 12.x) 2024 年 11 月 28 日 2025 年 11 月 28 日
2.12 CPU のみ 2024 年 6 月 30 日 2025 年 6 月 30 日
2.12 GPU(CUDA 11.x) 2024 年 6 月 30 日 2025 年 6 月 30 日
2.11 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.11 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.10 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.10 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.9 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.9 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.8 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.8 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.7 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.7 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.6 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.6 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.5 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.5 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.4 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.4 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.3 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.3 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.2 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.2 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.1 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.1 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
1.15 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
1.15 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日

最適化された TensorFlow ランタイム

次のコンテナ イメージでは、最適化された TensorFlow ランタイムを使用しています。詳細については、最適化された TensorFlow ランタイムを使用するをご覧ください。

使用可能な最適化された TensorFlow ランタイム コンテナ イメージ(クリックして展開)

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 サポートされているイメージ
ナイトリー CPU のみ 該当なし 該当なし
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.nightly:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.nightly:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.nightly:latest
ナイトリー GPU(CUDA 12.x) 該当なし 該当なし
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.nightly:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.nightly:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.nightly:latest
ナイトリー Cloud TPU 該当なし 該当なし
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.nightly:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.nightly:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.nightly:latest
2.17 CPU のみ 2024 年 7 月 11 日 2025 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-17:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-17:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-17:latest
2.17 GPU(CUDA 12.x) 2024 年 7 月 11 日 2025 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-17:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-17:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-17:latest
2.17 Cloud TPU 2024 年 7 月 11 日 2025 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-17:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-17:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-17:latest
2.16 CPU のみ 2024 年 4 月 26 日 2025 年 4 月 26 日
2.16 GPU(CUDA 12.x) 2024 年 4 月 26 日 2025 年 4 月 26 日
2.16 Cloud TPU 2024 年 4 月 26 日 2025 年 4 月 26 日
2.15 CPU のみ 2024 年 8 月 15 日 2025 年 8 月 15 日
2.15 GPU(CUDA 12.x) 2024 年 8 月 15 日 2025 年 8 月 15 日
2.15 Cloud TPU 2024 年 8 月 15 日 2025 年 8 月 15 日
2.14 CPU のみ 2024 年 8 月 15 日 2025 年 8 月 15 日
2.14 GPU(CUDA 12.x) 2024 年 8 月 15 日 2025 年 8 月 15 日
2.13 CPU のみ 2024 年 8 月 15 日 2025 年 8 月 15 日
2.13 GPU(CUDA 11.x) 2024 年 8 月 15 日 2025 年 8 月 15 日
2.12 CPU のみ 2024 年 5 月 15 日 2025 年 5 月 15 日
2.12 GPU(CUDA 11.x) 2024 年 5 月 15 日 2025 年 5 月 15 日
2.11 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.11 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.10 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.10 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.9 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.9 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
2.8 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
2.8 GPU(CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日

PyTorch

使用可能な PyTorch コンテナ イメージ(クリックして展開)

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 サポートされているイメージ
2.4(Python 3.9) CPU のみ 2026 年 7 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-4:latest
2.4(Python 3.9) GPU(CUDA 12.x) 2026 年 7 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-4:latest
2.4(Python 3.9) Cloud TPU 2026 年 7 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-4:latest
2.3(Python 3.9) CPU のみ 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-3:latest
2.3(Python 3.9) GPU(CUDA 12.x) 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-3:latest
2.3(Python 3.9) Cloud TPU 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-3:latest
2.2(Python 3.9) CPU のみ 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-2:latest
2.2(Python 3.9) GPU(CUDA 12.x) 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-2:latest
2.2(Python 3.9) Cloud TPU 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-2:latest
2.1(Python 3.9) CPU のみ 2024 年 12 月 1 日 2025 年 12 月 1 日
2.1(Python 3.9) GPU(CUDA 12.x) 2024 年 12 月 1 日 2025 年 12 月 1 日
2.1(Python 3.9) Cloud TPU 2024 年 12 月 1 日 2025 年 12 月 1 日
2.0(Python 3.9) CPU のみ 2024 年 7 月 27 日 2025 年 7 月 27 日
2.0(Python 3.9) GPU(CUDA 11.x) 2024 年 7 月 27 日 2025 年 7 月 27 日
1.13(Python 3.8) CPU のみ 2024 年 5 月 15 日 2025 年 5 月 15 日
1.13(Python 3.8) GPU(CUDA 11.x) 2024 年 5 月 15 日 2025 年 5 月 15 日
1.12 CPU のみ 2024 年 5 月 15 日 2025 年 5 月 15 日
1.12 GPU(CUDA 11.x) 2024 年 5 月 15 日 2025 年 5 月 15 日
1.11 CPU のみ 2024 年 5 月 15 日 2025 年 5 月 15 日
1.11 GPU(CUDA 11.x) 2024 年 5 月 15 日 2025 年 5 月 15 日

scikit-learn

使用可能な scikit-learn コンテナ イメージ(クリックして展開)

ML フレームワークのバージョン サポートされているアクセラレータ(該当する場合は CUDA のバージョン) パッチとサポート終了日 有効期間の終了 対応イメージ
1.6(Python 3.10) CPU のみ 2026 年 10 月 14 日 2027 年 10 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-6:latest
1.5(Python 3.10) CPU のみ 2026 年 7 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-5:latest
1.4(Python 3.10) CPU のみ 2026 年 1 月 14 日 2027 年 1 月 14 日
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-4:latest
1.3(Python 3.10) CPU のみ 2024 年 11 月 28 日 2025 年 11 月 28 日
1.2(Python 3.10) CPU のみ 2024 年 6 月 30 日 2025 年 6 月 30 日
1.0 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
0.24 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
0.23 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
0.22 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日
0.20 CPU のみ 2023 年 11 月 15 日 2024 年 11 月 15 日

XGBoost

使用可能な XGBoost コンテナ イメージ(クリックして展開)

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

サポートされている 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

パイプラインでサポートされているフレームワーク

Agent Platform Pipelines は、次の ML フレームワーク SDK バージョンをサポートしています。

これらのフレームワークは定期的に更新され、バグや脆弱性にパッチが適用されています。常に最新のパッチ バージョンを使用し、サポート終了日を過ぎたバージョンを使用しないようにすることをおすすめします。サポート終了日を過ぎたバージョンはカスタマー サポートの対象外となります。詳細については、Gemini Enterprise Agent Platform フレームワークのサポート ポリシーをご覧ください。

通常、パイプライン テンプレートを使用する場合は、次のガイドラインに従ってください。

  • ML フレームワークの最新パッチ バージョンを常に使用します。

  • 少なくとも 12 か月に 1 回、新規および既存のパイプライン テンプレートを ML フレームワークの最新のマイナー バージョンにアップグレードします。

  • ML フレームワークの新しいメジャー バージョンが利用可能になったら、24 か月以内に新規および既存のパイプライン テンプレートを新しいメジャー バージョンに移行します。

Kubeflow Pipelines SDK

KFP は、Agent Platform Pipelines でパイプラインを作成してオーケストレーションするために最もよく使用されるフレームワークです。Agent Platform Pipelines は、KFP を使用して pipeline_spec 中間表現(IR YAML 形式)v2.0 以降にコンパイルされたパイプライン定義を実行できます。

Agent Platform Pipelines は、次のバージョンの KFP SDK をサポートしています。

フレームワーク リリース日 サポート終了日
KFP SDK 2 2023 年 6 月 20 日 該当なし
KFP SDK 1.8 2021 年 9 月 3 日 2024 年 12 月 20 日

最新バージョンの KFP SDK では、サポート終了が宣言されている日にサポート終了の Python バージョンのサポートを終了します。唯一の例外は Python 3.7 で、2024 年 4 月 23 日にサポートを終了します。

TensorFlow Extended(TFX)SDK

TensorFlow では、TensorFlow Extended(TFX)SDK の新しいマイナー バージョンが四半期ごとにリリースされます。以前のバージョンの SDK からコンパイルされたパイプラインは引き続き実行できますが、パッチを受け取るのは SDK の最新のマイナー バージョンのみです。したがって、常に最新のマイナー バージョンを使用することをおすすめします。

フレームワーク リリース日 サポート終了日
TensorFlow Extended SDK 1.15 2024 年 4 月 29 日 2025 年 4 月 29 日
TensorFlow Extended SDK 1.14 2023 年 9 月 6 日 2024 年 9 月 6 日
TensorFlow Extended SDK 1.13 2023 年 5 月 3 日 2024 年 5 月 3 日
TensorFlow Extended SDK 1.12 2022 年 12 月 19 日 2023 年 12 月 19 日
TensorFlow Extended SDK 1.11 2022 年 11 月 23 日 2023 年 11 月 23 日
TensorFlow Extended SDK 1.10 2022 年 10 月 4 日 2023 年 10 月 4 日
TensorFlow Extended SDK 1.9 2022 年 7 月 15 日 2023 年 7 月 15 日
TensorFlow Extended SDK 1.8 2022 年 5 月 25 日 2023 年 5 月 25 日

Google Cloud パイプライン コンポーネント

Agent Platform Pipelines は、KFP 定義をサポートするために、次の Python パッケージとコンテナ イメージを公開し、定期的にパッチを適用しています。

フレームワーク リリース日 パッチとサポート終了日 有効期間の終了日 サポートされているイメージ
Google Cloud パイプライン コンポーネント 2 2023 年 6 月 20 日 該当なし 該当なし gcr.io/ml-pipeline/google-cloud-pipeline-components:latest
Google Cloud パイプライン コンポーネント 1 2022 年 2 月 25 日 2024 年 12 月 20 日 2025 年 6 月 20 日 gcr.io/ml-pipeline/google-cloud-pipeline-components:1.0.*

Google Cloud パイプライン コンポーネント SDK の最新バージョンは、サポート終了が宣言された日にサポート終了の Python バージョンのサポートを終了します。唯一の例外は Python 3.7 で、2024 年 4 月 23 日にサポートを終了します。

次のステップ