Daftar framework yang didukung

Halaman ini berisi daftar framework yang didukung dan paket konstituennya untuk layanan berikut:

Untuk penjelasan mendalam tentang kebijakan dukungan framework, lihat Kebijakan dukungan framework Vertex AI.

Framework yang didukung untuk pelatihan kustom

Vertex AI menggunakan image container untuk mengonfigurasi resource cloud guna melayani permintaan pelatihan Anda. Versi berikut didukung untuk container bawaan pelatihan khusus:

TensorFlow

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
2.15 (Python 3.10) Khusus TPU 23 Oktober 2024 23 Oktober 2025
  • 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) Khusus CPU 26 Sep 2024 26 September 2025
  • 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) 26 Sep 2024 26 September 2025
  • 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) Khusus CPU 5 Juli 2024 5 Juli 2025
  • 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) 5 Juli 2024 5 Juli 2025
  • 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) Khusus CPU 30 Jun 2024 30 Jun 2025
  • 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) 30 Jun 2024 30 Jun 2025
  • 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) Khusus TPU 30 Jun 2024 30 Jun 2025
  • 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) Khusus CPU 15 Nov 2023 15 Nov 2024
  • 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) 15 Nov 2023 15 Nov 2024
  • 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) Khusus CPU 15 Nov 2023 15 Nov 2024
  • 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) 15 Nov 2023 15 Nov 2024
  • 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) Khusus CPU 15 Nov 2023 15 Nov 2024
  • 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) 15 Nov 2023 15 Nov 2024
  • 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) Khusus CPU 15 Nov 2023 15 Nov 2024
  • 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) 15 Nov 2023 15 Nov 2024
  • 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) Khusus CPU 15 Nov 2023 15 Nov 2024
  • 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) 15 Nov 2023 15 Nov 2024
  • 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) Khusus CPU 15 Nov 2023 15 Nov 2024
  • 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) 15 Nov 2023 15 Nov 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
2.3 (Python 3.10) Khusus CPU 24 April 2025 24 April 2026
  • 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) Khusus GPU (CUDA 12.x) 24 April 2025 24 April 2026
  • 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) Khusus CPU 30 Jan 2025 23 Oktober 2026
  • 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) Khusus GPU (CUDA 12.x) 30 Jan 2025 30 Jan 2026
  • 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.1 (Python 3.10) Khusus CPU 4 Oktober 2024 4 Oktober 2025
  • 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) Khusus GPU (CUDA 12.x) 4 Oktober 2024 4 Oktober 2025
  • 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) Khusus TPU 23 Oktober 2024 23 Oktober 2025
  • 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) Khusus CPU 15 Mar 2024 15 Mar 2025
  • 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) Khusus GPU (CUDA 11.x) 15 Mar 2024 15 Mar 2025
  • 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) Khusus TPU 15 Mei 2024 15 Mei 2025
  • 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 atau GPU (CUDA 11.x) 15 Mei 2024 15 Mei 2025
  • 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 atau GPU (CUDA 11.x) 8 Des 2023 8 Des 2024
  • 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 atau GPU (CUDA 11.x) 15 Mei 2024 15 Mei 2025
  • 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 atau GPU (CUDA 11.x) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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) 1 Sep 2023 1 Sep 2024
  • 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

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
1.0 (Python 3.10) Khusus CPU 30 Juni 2024 30 Juni 2025
  • 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) 30 Juni 2024 30 Juni 2025
  • 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) Khusus CPU 1 Sep 2023 1 Sep 2024
  • 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

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
1.6 (Python 3.10) Khusus CPU 30 Juni 2024 30 Juni 2025
  • 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) 30 Juni 2024 30 Juni 2025
  • 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) Khusus CPU 15 Nov 2023 15 Nov 2024
  • 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

Framework yang didukung untuk prediksi dan penjelasan

Vertex AI menggunakan image container untuk mengonfigurasi resource cloud guna melayani permintaan prediksi Anda. Versi berikut didukung pada container bawaan untuk prediksi dan penjelasan:

TensorFlow

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
2.13 Khusus CPU 28 November 2024 28 November 2025
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-13:latest
2.13 GPU (CUDA 12.x) 28 November 2024 28 November 2025
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-13:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-13:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-13:latest
2.12 Khusus CPU 30 Juni 2024 30 Juni 2025
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-12:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-12:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-12:latest
2.12 GPU (CUDA 11.x) 30 Juni 2024 30 Juni 2025
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-12:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-12:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-12:latest
2.11 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-11:latest
2.11 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-11:latest
2.10 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-10:latest
2.10 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-10:latest
2.9 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-9:latest
2.9 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-9:latest
2.8 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-8:latest
2.8 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-8:latest
2,7 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-7:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-7:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-7:latest
2,7 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-7:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-7:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-7:latest
2,6 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-6:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-6:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-6:latest
2,6 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-6:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-6:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-6:latest
2,5 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-5:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-5:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-5:latest
2,5 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-5:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-5:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-5:latest
2,4 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-4:latest
2,4 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-4:latest
2.3 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-3:latest
2.3 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-3:latest
2.2 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-2:latest
2.2 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest
2.1 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-1:latest
2.1 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf2-gpu.2-2:latest
1.15 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf-cpu.1-15:latest
1.15 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/tf-gpu.1-15:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/tf-gpu.1-15:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/tf-gpu.1-15:latest

Runtime TensorFlow yang dioptimalkan

Image container berikut menggunakan runtime TensorFlow yang dioptimalkan. Untuk mengetahui informasi lebih lanjut, lihat Menggunakan runtime TensorFlow yang dioptimalkan.

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
tiap hari Khusus CPU Tidak berlaku Tidak berlaku
  • 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
tiap hari GPU (CUDA 12.x) Tidak berlaku Tidak berlaku
  • 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
tiap hari Cloud TPU Tidak berlaku Tidak berlaku
  • 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 Khusus CPU 11 Juli 2024 11 Juli 2025
  • 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) 11 Juli 2024 11 Juli 2025
  • 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 11 Juli 2024 11 Juli 2025
  • 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 Khusus CPU 26 April 2024 26 April 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-16:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-16:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-16:latest
2,16 GPU (CUDA 12.x) 26 April 2024 26 April 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-16:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-16:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-16:latest
2,16 Cloud TPU 26 April 2024 26 April 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-16:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-16:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-16:latest
2,15 Khusus CPU 15 Agu 2024 15 Agu 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-15:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-15:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-15:latest
2,15 GPU (CUDA 12.x) 15 Agu 2024 15 Agu 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-15:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-15:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-15:latest
2,15 Cloud TPU 15 Agu 2024 15 Agu 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-15:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-15:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-tpu.2-15:latest
2,14 Khusus CPU 15 Agu 2024 15 Agu 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-14:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-14:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-14:latest
2,14 GPU (CUDA 12.x) 15 Agu 2024 15 Agu 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-14:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-14:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-14:latest
2.13 Khusus CPU 15 Agu 2024 15 Agu 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-13:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-13:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-13:latest
2.13 GPU (CUDA 11.x) 15 Agu 2024 15 Agu 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-13:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-13:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-13:latest
2.12 Khusus CPU 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-12:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-12:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-12:latest
2.12 GPU (CUDA 11.x) 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-12:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-12:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-12:latest
2.11 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-11:latest
2.11 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-11:latest
2.10 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-10:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-10:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-10:latest
2.10 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-10:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-10:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-10:latest
2.9 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-9:latest
2.9 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-9:latest
2.8 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-cpu.2-8:latest
2.8 GPU (CUDA 11.x) 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai-restricted/prediction/tf_opt-gpu.2-8:latest

PyTorch

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
2.1 (Python 3.9) Khusus CPU 1 Des 2024 01 Des 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-1:latest
2.1 (Python 3.9) GPU (CUDA 12.x) 1 Des 2024 01 Des 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-1:latest
2.1 (Python 3.9) Cloud TPU 1 Des 2024 01 Des 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-tpu.2-1:latest
2.0 (Python 3.9) Khusus CPU 27 Juli 2024 27 Juli 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.2-0:latest
2.0 (Python 3.9) GPU (CUDA 11.x) 27 Juli 2024 27 Juli 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.2-0:latest
1.13 (Python 3.8) Khusus CPU 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-13:latest
1.13 (Python 3.8) GPU (CUDA 11.x) 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-13:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-13:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-13:latest
1.12 Khusus CPU 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-12:latest
1.12 GPU (CUDA 11.x) 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-12:latest
1.11 Khusus CPU 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-cpu.1-11:latest
1.11 GPU (CUDA 11.x) 15 Mei 2024 15 Mei 2025
  • us-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-11:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-11:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/pytorch-gpu.1-11:latest

scikit-learn

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
1.3 (Python 3.10) Khusus CPU 28 November 2024 28 November 2025
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-3:latest
1.2 (Python 3.10) Khusus CPU 30 Juni 2024 30 Juni 2025
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-2:latest
1.0 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-0:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-0:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.1-0:latest
0,24 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-24:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-24:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-24:latest
0,23 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-23:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-23:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-23:latest
0,22 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-22:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-22:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-22:latest
0,20 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-20:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-20:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/sklearn-cpu.0-20:latest

XGBoost

Versi framework ML Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
1.7 (Python 3.10) Khusus CPU 30 Juni 2024 30 Juni 2025
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest
1.6 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-6:latest
1,5 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-5:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-5:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-5:latest
1.4 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-4:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-4:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-4:latest
1,3 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-3:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-3:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-3:latest
1,2 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-2:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-2:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-2:latest
1.1 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-1:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-1:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-1:latest
0.90 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-90:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-90:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-90:latest
0.82 Khusus CPU 15 Nov 2023 15 Nov 2024
  • us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-82:latest
  • europe-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-82:latest
  • asia-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.0-82:latest

Versi Ray di Vertex AI yang didukung

Image container Ray on Vertex AI tersedia. Versi penampung bawaan berikut didukung:

Versi Ray Akselerator yang didukung (dan versi CUDA, jika berlaku) Akhir patch dan tanggal dukungan Akhir ketersediaan Image yang didukung
2.33.0 (Python 3.10) Khusus CPU 16 September 2025 16 September 2026
  • 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) 16 September 2025 16 September 2026
  • 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) Khusus CPU 7 Mei 2025 7 Mei 2026
  • 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) 7 Mei 2025 7 Mei 2026
  • 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

Framework yang didukung untuk pipeline

Vertex AI Pipelines mendukung versi SDK framework ML berikut:

Karena framework ini diupdate secara berkala untuk mem-patch bug dan kerentanan, sebaiknya Anda selalu menggunakan versi patch terbaru dan jangan melampaui tanggal dukungan. Dukungan pelanggan tidak diberikan jika versi melampaui tanggal dukungan. Untuk mengetahui informasi lebih lanjut, lihat kebijakan dukungan framework Vertex AI.

Secara umum, ikuti panduan ini saat menggunakan template pipeline:

  • Dapatkan info terkini tentang versi patch framework ML terbaru.

  • Upgrade template pipeline baru dan yang sudah ada ke versi minor terbaru dari framework ML setidaknya sekali setiap 12 bulan.

  • Setelah versi utama framework ML yang baru tersedia, migrasikan template pipeline baru dan yang sudah ada ke versi utama baru dalam waktu 24 bulan.

Kubeflow Pipelines SDK

KFP adalah framework yang paling umum digunakan untuk menyusun dan mengorkestrasi pipeline di Vertex AI Pipelines. Vertex AI Pipelines dapat menjalankan definisi pipeline yang dikompilasi menggunakan KFP ke dalam pipeline_spec Intermediate Representation (format IR YAML) v2.0 atau yang lebih baru.

Vertex AI Pipelines mendukung versi KFP SDK berikut:

Framework Tanggal rilis Akhir tanggal dukungan
SDK KFP 2 20 Juni 2023 Tidak berlaku
SDK KFP 1.8 3 Sep 2021 20 Des 2024

Versi terbaru KFP SDK akan menghentikan dukungan untuk versi Python yang tidak digunakan lagi pada hari versi tersebut dinyatakan tidak digunakan lagi. Satu-satunya pengecualian untuk hal ini adalah Python 3.7, yang dukungannya akan dihentikan pada 23 April 2024.

SDK TensorFlow Extended (TFX)

TensorFlow merilis versi minor baru dari TensorFlow Extended (TFX) SDK setiap kuartal. Anda masih dapat menjalankan pipeline yang dikompilasi dari SDK versi lama, tetapi hanya SDK versi minor terbaru yang menerima patch. Oleh karena itu, sebaiknya Anda terus menggunakan versi minor terbaru.

Framework Tanggal rilis Akhir tanggal dukungan
TensorFlow Extended SDK 1.15 29 April 2024 29 April 2025
TensorFlow Extended SDK 1.14 6 September 2023 6 September 2024
TensorFlow Extended SDK 1.13 3 Mei 2023 3 Mei 2024
TensorFlow Extended SDK 1.12 19 Des 2022 19 Des 2023
TensorFlow Extended SDK 1.11 23 Nov 2022 23 Nov 2023
TensorFlow Extended SDK 1.10 4 Okt 2022 4 Okt 2023
TensorFlow Extended SDK 1.9 15 Juli 2022 15 Juli 2023
TensorFlow Extended SDK 1.8 25 Mei 2022 25 Mei 2023

Google Cloud Pipeline Components

Vertex AI Pipelines memublikasikan dan mem-patch paket Python dan image container berikut secara rutin untuk membantu dalam definisi KFP:

Framework Tanggal rilis Akhir patch dan tanggal dukungan Akhir tanggal ketersediaan Image yang didukung
Google Cloud Pipeline Components 2 20 Juni 2023 Tidak berlaku Tidak berlaku gcr.io/ml-pipeline/google-cloud-pipeline-components:latest
Google Cloud Pipeline Components 1 25 Feb 2022 20 Des 2024 20 Jun 2025 gcr.io/ml-pipeline/google-cloud-pipeline-components:1.0.*

Versi terbaru Google Cloud Pipeline Components SDK akan menghentikan dukungan untuk versi Python yang tidak digunakan lagi pada hari versi tersebut dinyatakan tidak digunakan lagi. Satu-satunya pengecualian untuk hal ini adalah Python 3.7, yang dukungannya akan dihentikan pada 23 April 2024.

Framework yang didukung untuk notebook yang dikelola pengguna Vertex AI Workbench

Notebook yang dikelola pengguna Vertex AI Workbench mendukung versi framework ML dalam tabel berikut. Untuk menemukan VERSION_DATE spesifik untuk image, lihat Mencantumkan versi yang tersedia.

Versi dasar

Versi framework ML Versi patch saat ini Akselerator yang didukung Akhir patch dan tanggal dukungan Akhir tanggal ketersediaan Nama kelompok image
Base-CPU (Python 3.10/Debian 11) Tidak berlaku (T/A) Khusus CPU 1 Juli 2024 1 Juli 2025 common-cpu-VERSION_DATE-debian-11
Base-cu122 (Python 3.10) CUDA 12.2 GPU (CUDA 12.2) 8 Jan 2025 8 Januari 2026 common-cu122-VERSION_DATE-debian-11-py310
Base-cu121 (Python 3.10) CUDA 12.1 GPU (CUDA 12.1) 28 Feb 2024 28 Februari 2025 common-cu121-VERSION_DATE-debian-11-py310
Base-cu118 (Python 3.10) CUDA 11.8 GPU (CUDA 11.8) 1 Juli 2024 1 Juli 2025 common-cu118-VERSION_DATE-debian-11-py310
Base-cu113 (Python 3.10) CUDA 11.3 GPU (CUDA 11.3) 1 Jan 2024 1 Jan 2025 common-cu113-VERSION_DATE-debian-11-py310
Base-cu113 (Python 3.7) CUDA 11.3 GPU (CUDA 11.3) 18 Sep 2023 18 Sep 2024 common-cu113-VERSION_DATE-py37
Base-cu110 (Python 3.7) CUDA 11.0 GPU (CUDA 11.0) 18 Sep 2023 18 Sep 2024 common-cu110-VERSION_DATE-py37
Base-CPU (Python 3.7) Tidak berlaku (T/A) Khusus CPU 18 Sep 2023 18 Sep 2024 common-cpu-VERSION_DATE-debian-10

Versi TensorFlow

Versi framework ML Versi patch saat ini Akselerator yang didukung Akhir patch dan tanggal dukungan Akhir tanggal ketersediaan Nama kelompok image
2.17 (Python 3.10) 2.17.0 Khusus CPU 11 Juli 2025 11 Juli 2026 tf-2-17-cpu-VERSION_DATE-py310
2.17 (Python 3.10) 2.17.0 GPU (CUDA 12.3) 11 Juli 2025 11 Juli 2026 tf-2-17-cu123-VERSION_DATE-py310
2.16 (Python 3.10) 2.16.2 Khusus CPU 28 Jun 2025 28 Juni 2026 tf-2-16-cpu-VERSION_DATE-py310
2.16 (Python 3.10) 2.16.2 GPU (CUDA 12.3) 28 Jun 2025 28 Juni 2026 tf-2-16-cu123-VERSION_DATE-py310
2.15 (Python 3.10) 2.15.0 Khusus CPU 14 November 2024 14 November 2025 tf-2-15-cpu-VERSION_DATE-py310
2.15 (Python 3.10) 2.15.0 GPU (CUDA 12.1) 14 November 2024 14 November 2025 tf-2-15-cu121-VERSION_DATE-py310
2.14 (Python 3.10) 2.14.0 Khusus CPU 26 Sep 2024 26 September 2025 tf-2-14-cpu-VERSION_DATE-py310
2.14 (Python 3.10) 2.14.0 GPU (CUDA 11.8) 26 Sep 2024 26 September 2025 tf-2-14-cu118-VERSION_DATE-py310
2.13 (Python 3.10) 2.13.0 Khusus CPU 5 Juli 2024 5 Juli 2025 tf-2-13-cpu-VERSION_DATE-py310
2.13 (Python 3.10) 2.13.0 GPU (CUDA 11.8) 5 Juli 2024 5 Juli 2025 tf-2-13-cu118-VERSION_DATE-py310
2.12 (Python 3.10) 2.12.0 Khusus CPU 30 Juni 2024 30 Juni 2025 tf-2-12-cpu-VERSION_DATE-py310
2.12 (Python 3.10) 2.12.0 GPU (CUDA 11.8) 18 Jan 2024 18 Jan 2025 tf-2-12-cu113-VERSION_DATE-py310
2.11 (Python 3.10) 2.11.0 Khusus CPU 15 Nov 2023 15 Nov 2024 tf-2-11-cpu-VERSION_DATE-py310
2.11 (Python 3.10) 2.11.0 GPU (CUDA 11.3) 15 Nov 2023 15 Nov 2024 tf-2-11-cu113-VERSION_DATE-py310
2.11 2.11.0 Khusus CPU 15 Nov 2023 15 Nov 2024 tf-2-11-cpu-VERSION_DATE-py37
2.11 2.11.0 GPU (CUDA 11.3) 15 Nov 2023 15 Nov 2024 tf-2-11-cu113-VERSION_DATE-py37
2.10 2.10.1 Khusus CPU 15 Nov 2023 15 Nov 2024 tf-2-10-cpu-VERSION_DATE-py37
2.10 2.10.1 GPU (CUDA 11.3) 15 Nov 2023 15 Nov 2024 tf-2-10-cu113-VERSION_DATE-py37
2.9 2.9.3 Khusus CPU 15 Nov 2023 15 Nov 2024 tf-2-9-cpu-VERSION_DATE-py37
2.9 2.9.3 GPU (CUDA 11.3) 15 Nov 2023 15 Nov 2024 tf-2-9-cu113-VERSION_DATE-py37
2.8 2.8.4 Khusus CPU 15 Nov 2023 15 Nov 2024 tf-2-8-cpu-VERSION_DATE-py37
2.8 2.8.4 GPU (CUDA 11.3) 15 Nov 2023 15 Nov 2024 tf-2-8-cu113-VERSION_DATE-py37
2.6 (py39) 2.6.5 Khusus CPU 1 Sep 2023 1 Sep 2024 tf-2-6-cpu-VERSION_DATE-py39
2.6 (py39) 2.6.5 GPU (CUDA 11.3) 1 Sep 2023 1 Sep 2024 tf-2-6-cu110-VERSION_DATE-py39
2.6 (py37) 2.6.5 Khusus CPU 18 Sep 2023 18 Sep 2024 tf-2-6-cpu-VERSION_DATE-py37
2.6 (py37) 2.6.5 GPU (CUDA 11.3) 18 Sep 2023 18 Sep 2024 tf-2-6-cu110-VERSION_DATE-py37
2.3 2.3.4 Khusus CPU 1 Sep 2023 1 Sep 2024 tf-2-3-cpu
2.3 2.3.4 GPU (CUDA 11.3) 1 Sep 2023 1 Sep 2024 tf-2-3-cu110-VERSION_DATE

Versi PyTorch

Versi framework ML Versi patch saat ini Akselerator yang didukung Akhir patch dan tanggal dukungan Akhir tanggal ketersediaan Nama kelompok image
2.2 (Python 3.10) 2.2.0 CUDA 12.1 30 Jan 2025 30 Jan 2026 pytorch-2-2-VERSION_DATE-py310
2.1 (Python 3.10) 2.1.0 CUDA 12.1 4 Oktober 2024 4 Oktober 2025 pytorch-2-1-VERSION_DATE-py310
2.0 (Python 3.10) 2.0.0 CUDA 11.8 15 Mar 2024 15 Mar 2025 pytorch-2-0-VERSION_DATE-py310
1.13 (Python 3.10) 1.13.1 CUDA 11.3 8 Des 2023 8 Des 2024 pytorch-1-13-VERSION_DATE-py310
1.13 1.13.1 CUDA 11.3 8 Des 2023 8 Des 2024 pytorch-1-13-VERSION_DATE-py37
1.12 1.12.1 CUDA 11.3 1 Sep 2023 1 Sep 2024 pytorch-1-12-VERSION_DATE-py310

Langkah berikutnya