支持的框架列表

本页面列出了以下服务支持的框架及相应框架的组成软件包:

如需深入了解框架支持政策,请参阅 Vertex AI 框架支持政策

用于自定义训练的支持框架

Vertex AI 使用容器映像来配置云资源,以便处理您的训练请求。自定义训练预构建容器支持以下版本:

TensorFlow

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
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

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
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

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
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

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
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 使用容器映像来配置云资源,以便处理您的预测请求。用于预测和说明的预构建容器支持以下版本:

TensorFlow

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
2.13 仅 CPU 2024 年 11 月 28 日 2025 年 11 月 28 日
  • 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) 2024 年 11 月 28 日 2025 年 11 月 28 日
  • 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 仅 CPU 2024 年 6 月 30 日 2025 年 6 月 30 日
  • 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) 2024 年 6 月 30 日 2025 年 6 月 30 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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

优化的 TensorFlow 运行时

以下容器映像使用优化的 TensorFlow 运行时。如需了解详情,请参阅使用优化的 TensorFlow 运行时

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
Nightly 仅 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
Nightly 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
Nightly 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.15 仅 CPU 2024 年 8 月 15 日 2025 年 8 月 15 日
  • 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) 2024 年 8 月 15 日 2025 年 8 月 15 日
  • 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 2024 年 8 月 15 日 2025 年 8 月 15 日
  • 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 仅 CPU 2024 年 8 月 15 日 2025 年 8 月 15 日
  • 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) 2024 年 8 月 15 日 2025 年 8 月 15 日
  • 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 仅 CPU 2024 年 8 月 15 日 2025 年 8 月 15 日
  • 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) 2024 年 8 月 15 日 2025 年 8 月 15 日
  • 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 仅 CPU 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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) 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
2.1 (Python 3.9) 仅 CPU 2024 年 12 月 1 日 2025 年 12 月 1 日
  • 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) 2024 年 12 月 1 日 2025 年 12 月 1 日
  • 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 2024 年 12 月 1 日 2025 年 12 月 1 日
  • 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) 仅 CPU 2024 年 7 月 27 日 2025 年 7 月 27 日
  • 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) 2024 年 7 月 27 日 2025 年 7 月 27 日
  • 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) 仅 CPU 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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) 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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 仅 CPU 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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) 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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 仅 CPU 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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) 2024 年 5 月 15 日 2025 年 5 月 15 日
  • 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

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
1.3 (Python 3.10) 仅 CPU 2024 年 11 月 28 日 2025 年 11 月 28 日
  • 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) 仅 CPU 2024 年 6 月 30 日 2025 年 6 月 30 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
1.7 (Python 3.10) 仅 CPU 2024 年 6 月 30 日 2025 年 6 月 30 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • 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

用于流水线的支持框架

Vertex AI Pipelines 支持以下机器学习框架 SDK 版本:

由于这些框架会定期更新以修补错误和漏洞,因此建议您始终使用最新的补丁版本并保持在支持终止期限内。如果某个版本超过支持终止日期,则无法享受客户支持服务。如需了解详情,请参阅 Vertex AI 框架支持政策

通常,请在使用流水线模板时遵循以下准则:

  • 及时了解机器学习框架的最新补丁版本。

  • 至少每 12 个月将新的和现有的流水线模板升级到机器学习框架的最新次要版本一次。

  • 机器学习框架推出新的主要版本后,请在 24 个月内将新的和现有的流水线模板迁移到新的主要版本。

Kubeflow Pipelines SDK

KFP 是在 Vertex AI Pipelines 中编写和编排流水线的较常用框架。Vertex AI Pipelines 可以执行使用 KFP 编译为 pipeline_spec 中间表示法(IR YAML 格式)v2.0 或更高版本的流水线定义。

Vertex AI 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.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 流水线组件

Vertex AI Pipelines 发布并定期修补以下 Python 软件包和容器映像,以帮助执行 KFP 定义

框架 发布日期 补丁程序和支持日期结束 提供终止日期 支持的图片
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 日停止。

Vertex AI Workbench 用户管理的笔记本支持的框架

Vertex AI Workbench 用户管理的笔记本支持以下表中的机器学习框架版本。如需查找映像的特定 VERSION_DATE,请参阅列出可用的版本

基础版本

机器学习框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
Base-CPU (Python 3.10 / Debian 11) 不适用 (N/A) 仅 CPU 2024 年 7 月 1 日 2025 年 7 月 1 日 common-cpu-VERSION_DATE-debian-11
Base-cu121 (Python 3.10) CUDA 12.1 GPU (CUDA 12.1) 2024 年 2 月 28 日 2025 年 2 月 28 日 common-cu121-VERSION_DATE-debian-11-py310
Base-cu118 (Python 3.10) CUDA 11.8 GPU (CUDA 11.8) 2024 年 7 月 1 日之后 2025 年 7 月 1 日之后 common-cu118-VERSION_DATE-debian-11-py310
Base-cu113 (Python 3.10) CUDA 11.3 GPU (CUDA 11.3) 2024 年 1 月 1 日 2025 年 1 月 1 日 common-cu113-VERSION_DATE-debian-11-py310
Base-cu113 (Python 3.7) CUDA 11.3 GPU (CUDA 11.3) 2023 年 9 月 18 日 2024 年 9 月 18 日 common-cu113-VERSION_DATE-py37
Base-cu110 (Python 3.7) CUDA 11.0 GPU (CUDA 11.0) 2023 年 9 月 18 日 2024 年 9 月 18 日 common-cu110-VERSION_DATE-py37
Base-CPU (Python 3.7) 不适用 (N/A) 仅 CPU 2023 年 9 月 18 日 2024 年 9 月 18 日 common-cpu-VERSION_DATE-debian-10

TensorFlow 版本

机器学习框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
2.15 (Python 3.10) 2.15.0 仅 CPU 2024 年 11 月 14 日 2025 年 11 月 14 日 tf-2-15-cpu-VERSION_DATE-py310
2.15 (Python 3.10) 2.15.0 GPU (CUDA 12.1) 2024 年 11 月 14 日 2025 年 11 月 14 日 tf-2-15-cu121-VERSION_DATE-py310
2.14 (Python 3.10) 2.14.0 仅 CPU 2024 年 9 月 26 日 2025 年 9 月 26 日 tf-2-14-cpu-VERSION_DATE-py310
2.14 (Python 3.10) 2.14.0 GPU (CUDA 11.8) 2024 年 9 月 26 日 2025 年 9 月 26 日 tf-2-14-cu118-VERSION_DATE-py310
2.13 (Python 3.10) 2.13.0 仅 CPU 2024 年 7 月 5 日 2025 年 7 月 5 日 tf-2-13-cpu-VERSION_DATE-py310
2.13 (Python 3.10) 2.13.0 GPU (CUDA 11.8) 2024 年 7 月 5 日 2025 年 7 月 5 日 tf-2-13-cu118-VERSION_DATE-py310
2.12 (Python 3.10) 2.12.0 仅 CPU 2024 年 6 月 30 日 2025 年 6 月 30 日 tf-2-12-cpu-VERSION_DATE-py310
2.12 (Python 3.10) 2.12.0 GPU (CUDA 11.8) 2024 年 1 月 18 日 2025 年 1 月 18 日 tf-2-12-cu113-VERSION_DATE-py310
2.11 (Python 3.10) 2.11.0 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-11-cpu-VERSION_DATE-py310
2.11 (Python 3.10) 2.11.0 GPU (CUDA 11.3) 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-11-cu113-VERSION_DATE-py310
2.11 2.11.0 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-11-cpu-VERSION_DATE-py37
2.11 2.11.0 GPU (CUDA 11.3) 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-11-cu113-VERSION_DATE-py37
2.10 2.10.1 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-10-cpu-VERSION_DATE-py37
2.10 2.10.1 GPU (CUDA 11.3) 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-10-cu113-VERSION_DATE-py37
2.9 2.9.3 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-9-cpu-VERSION_DATE-py37
2.9 2.9.3 GPU (CUDA 11.3) 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-9-cu113-VERSION_DATE-py37
2.8 2.8.4 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-8-cpu-VERSION_DATE-py37
2.8 2.8.4 GPU (CUDA 11.3) 2023 年 11 月 15 日 2024 年 11 月 15 日 tf-2-8-cu113-VERSION_DATE-py37
2.6 (py39) 2.6.5 仅 CPU 2024 年 8 月 10 日 2025 年 8 月 10 日 tf-2-6-cpu-VERSION_DATE-py39
2.6 (py39) 2.6.5 GPU (CUDA 11.3) 2024 年 8 月 10 日 2025 年 8 月 10 日 tf-2-6-cu110-VERSION_DATE-py39
2.6 (py37) 2.6.5 仅 CPU 2023 年 9 月 18 日 2024 年 9 月 18 日 tf-2-6-cpu-VERSION_DATE-py37
2.6 (py37) 2.6.5 GPU (CUDA 11.3) 2023 年 9 月 18 日 2024 年 9 月 18 日 tf-2-6-cu110-VERSION_DATE-py37
2.3 2.3.4 仅 CPU 2023 年 9 月 1 日 2024 年 9 月 1 日 tf-2-3-cpu
2.3 2.3.4 GPU (CUDA 11.3) 2023 年 9 月 1 日 2024 年 9 月 1 日 tf-2-3-cu110-VERSION_DATE

PyTorch 版本

机器学习框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
2.2 (Python 3.10) 2.2.0 CUDA 12.1 2025 年 1 月 30 日 2026 年 1 月 30 日 pytorch-2-2-VERSION_DATE-py310
2.1 (Python 3.10) 2.1.0 CUDA 12.1 2024 年 10 月 4 日 2025 年 10 月 4 日 pytorch-2-1-VERSION_DATE-py310
2.0 (Python 3.10) 2.0.0 CUDA 11.8 2024 年 3 月 15 日 2025 年 3 月 15 日 pytorch-2-0-VERSION_DATE-py310
1.13 (Python 3.10) 1.13.1 CUDA 11.3 2023 年 12 月 8 日 2024 年 12 月 8 日 pytorch-1-13-VERSION_DATE-py310
1.13 1.13.1 CUDA 11.3 2023 年 12 月 8 日 2024 年 12 月 8 日 pytorch-1-13-VERSION_DATE-py37
1.12 1.12.1 CUDA 11.3 2023 年 9 月 1 日 2024 年 9 月 1 日 pytorch-1-12-VERSION_DATE-py310

后续步骤