用於自訂訓練工作的預先建構容器

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 Nov 14, 2024 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) Nov 14, 2024 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 Nov 6, 2024 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 Oct 4, 2024 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

Vertex AI 支援的 Ray 版本

Ray on Vertex AI 容器映像檔現已推出。 系統支援下列預先建構的容器版本:

Ray 版本 支援的加速器 (如適用,請提供 CUDA 版本) 修補程式和支援服務終止日期 供應期限 支援的圖片
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 欄位。

後續步驟