本页面列出了以下服务支持的框架及相应框架的组成软件包:
如需深入了解框架支持政策,请参阅 Vertex AI 框架支持政策。
用于自定义训练的支持框架
Vertex AI 使用容器映像来配置云资源,以便处理您的训练请求。自定义训练预构建容器支持以下版本:
TensorFlow
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
2.15 (Python 3.10) | 仅限 TPU | 2024 年 10 月 23 日 | 2025 年 10 月 23 日 |
|
2.14 (Python 3.10) | 仅 CPU | 2024 年 9 月 26 日 | 2025 年 9 月 26 日 |
|
2.14 (Python 3.10) | GPU (CUDA 11.x) | 2024 年 9 月 26 日 | 2025 年 9 月 26 日 |
|
2.13 (Python 3.10) | 仅 CPU | 2024 年 7 月 5 日 | 2025 年 7 月 5 日 |
|
2.13 (Python 3.10) | GPU (CUDA 11.x) | 2024 年 7 月 5 日 | 2025 年 7 月 5 日 |
|
2.12 (Python 3.10) | 仅 CPU | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
2.12 (Python 3.10) | GPU (CUDA 11.x) | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
2.12 (Python 3.8) | 仅限 TPU | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
2.11 (Python 3.10) | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.11 (Python 3.10) | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.11 (Python 3.7) | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.11 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 (Python 3.10) | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 (Python 3.10) | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 (Python 3.7) | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 (Python 3.10) | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 (Python 3.10) | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 (Python 3.7) | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.7 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.7 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.6 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.6 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.5 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.5 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.4 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.4 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.3 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.3 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.2 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.2 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.1 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
2.1 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.15 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.15 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
PyTorch
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
2.3 (Python 3.10) | 仅 CPU | 2025 年 4 月 24 日 | 2026 年 4 月 24 日 |
|
2.3 (Python 3.10) | 仅限 GPU (CUDA 12.x) | 2025 年 4 月 24 日 | 2026 年 4 月 24 日 |
|
2.2 (Python 3.10) | 仅 CPU | 2025 年 1 月 30 日 | 2026 年 10 月 23 日 |
|
2.2 (Python 3.10) | 仅限 GPU (CUDA 12.x) | 2025 年 1 月 30 日 | 2026 年 1 月 30 日 |
|
2.1 (Python 3.10) | 仅 CPU | 2024 年 10 月 4 日 | 2025 年 10 月 4 日 |
|
2.1 (Python 3.10) | 仅限 GPU (CUDA 12.x) | 2024 年 10 月 4 日 | 2025 年 10 月 4 日 |
|
2.1 (Python 3.8) | 仅限 TPU | 2024 年 10 月 23 日 | 2025 年 10 月 23 日 |
|
2.0 (Python 3.10) | 仅 CPU | 2024 年 3 月 15 日 | 2025 年 3 月 15 日 |
|
2.0 (Python 3.10) | 仅限 GPU (CUDA 11.x) | 2024 年 3 月 15 日 | 2025 年 3 月 15 日 |
|
2.0 (Python 3.8) | 仅限 TPU | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.13 (Python 3.10) | CPU 或 GPU (CUDA 11.x) | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.13 (Python 3.7) | CPU 或 GPU (CUDA 11.x) | 2023 年 12 月 8 日 | 2024 年 12 月 8 日 |
|
1.12 (Python 3.10) | CPU 或 GPU (CUDA 11.x) | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.12 (Python 3.7) | CPU 或 GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.11 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.11 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.10 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.10 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.9 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.9 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.7 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.7 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.6 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.6 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.4 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
1.4 (Python 3.7) | GPU (CUDA 11.x) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
scikit-learn
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
1.0 (Python 3.10) | 仅 CPU | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
1.0 (Python 3.10) | GPU (CUDA 11.x) | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
0.23 (Python 3.7) | 仅 CPU | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 |
|
XGBoost
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
1.6 (Python 3.10) | 仅 CPU | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
1.6 (Python 3.10) | GPU (CUDA 11.x) | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
1.1 (Python 3.7) | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
用于预测和说明的支持框架
Vertex AI 使用容器映像来配置云资源,以便处理您的预测请求。用于预测和说明的预构建容器支持以下版本:
TensorFlow
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
2.13 | 仅 CPU | 2024 年 11 月 28 日 | 2025 年 11 月 28 日 |
|
2.13 | GPU (CUDA 12.x) | 2024 年 11 月 28 日 | 2025 年 11 月 28 日 |
|
2.12 | 仅 CPU | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
2.12 | GPU (CUDA 11.x) | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
2.11 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.11 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.10 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.10 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.7 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.7 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.6 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.6 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.5 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.5 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.4 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.4 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.3 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.3 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.2 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.2 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.1 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.1 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
1.15 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
1.15 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
优化的 TensorFlow 运行时
以下容器映像使用优化的 TensorFlow 运行时。如需了解详情,请参阅使用优化的 TensorFlow 运行时。
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
Nightly | 仅 CPU | 不适用 | 不适用 |
|
Nightly | GPU (CUDA 12.x) | 不适用 | 不适用 |
|
Nightly | Cloud TPU | 不适用 | 不适用 |
|
2.17 | 仅 CPU | 2024 年 7 月 11 日 | 2025 年 7 月 11 日 |
|
2.17 | GPU (CUDA 12.x) | 2024 年 7 月 11 日 | 2025 年 7 月 11 日 |
|
2.17 | Cloud TPU | 2024 年 7 月 11 日 | 2025 年 7 月 11 日 |
|
2.16 | 仅 CPU | 2024 年 4 月 26 日 | 2025 年 4 月 26 日 |
|
2.16 | GPU (CUDA 12.x) | 2024 年 4 月 26 日 | 2025 年 4 月 26 日 |
|
2.16 | Cloud TPU | 2024 年 4 月 26 日 | 2025 年 4 月 26 日 |
|
2.15 | 仅 CPU | 2024 年 8 月 15 日 | 2025 年 8 月 15 日 |
|
2.15 | GPU (CUDA 12.x) | 2024 年 8 月 15 日 | 2025 年 8 月 15 日 |
|
2.15 | Cloud TPU | 2024 年 8 月 15 日 | 2025 年 8 月 15 日 |
|
2.14 | 仅 CPU | 2024 年 8 月 15 日 | 2025 年 8 月 15 日 |
|
2.14 | GPU (CUDA 12.x) | 2024 年 8 月 15 日 | 2025 年 8 月 15 日 |
|
2.13 | 仅 CPU | 2024 年 8 月 15 日 | 2025 年 8 月 15 日 |
|
2.13 | GPU (CUDA 11.x) | 2024 年 8 月 15 日 | 2025 年 8 月 15 日 |
|
2.12 | 仅 CPU | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
2.12 | GPU (CUDA 11.x) | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
2.11 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.11 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.10 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.10 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.9 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
2.8 | GPU (CUDA 11.x) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
PyTorch
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
2.1 (Python 3.9) | 仅 CPU | 2024 年 12 月 1 日 | 2025 年 12 月 1 日 |
|
2.1 (Python 3.9) | GPU (CUDA 12.x) | 2024 年 12 月 1 日 | 2025 年 12 月 1 日 |
|
2.1 (Python 3.9) | Cloud TPU | 2024 年 12 月 1 日 | 2025 年 12 月 1 日 |
|
2.0 (Python 3.9) | 仅 CPU | 2024 年 7 月 27 日 | 2025 年 7 月 27 日 |
|
2.0 (Python 3.9) | GPU (CUDA 11.x) | 2024 年 7 月 27 日 | 2025 年 7 月 27 日 |
|
1.13 (Python 3.8) | 仅 CPU | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.13 (Python 3.8) | GPU (CUDA 11.x) | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.12 | 仅 CPU | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.12 | GPU (CUDA 11.x) | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.11 | 仅 CPU | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
1.11 | GPU (CUDA 11.x) | 2024 年 5 月 15 日 | 2025 年 5 月 15 日 |
|
scikit-learn
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
1.3 (Python 3.10) | 仅 CPU | 2024 年 11 月 28 日 | 2025 年 11 月 28 日 |
|
1.2 (Python 3.10) | 仅 CPU | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
1.0 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
0.24 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
0.23 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
0.22 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
0.20 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
XGBoost
机器学习框架版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
1.7 (Python 3.10) | 仅 CPU | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 |
|
1.6 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
1.5 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
1.4 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
1.3 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
1.2 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
1.1 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
0.90 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
0.82 | 仅 CPU | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 |
|
Ray on Vertex AI 支持的版本
Ray on Vertex AI 容器映像可供使用。支持以下版本的预构建容器:
Ray 版本 | 支持的加速器(以及 CUDA 版本,如适用) | 补丁程序和支持日期结束 | 提供终止日期 | 支持的图片 |
---|---|---|---|---|
2.33.0 (Python 3.10) | 仅 CPU | 2025 年 9 月 16 日 | 2026 年 9 月 16 日 |
|
2.33.0 (Python 3.10) | GPU (CUDA 11.x) | 2025 年 9 月 16 日 | 2026 年 9 月 16 日 |
|
2.9.3 (Python 3.10) | 仅 CPU | 2025 年 5 月 7 日 | 2026 年 5 月 7 日 |
|
2.9.3 (Python 3.10) | GPU (CUDA 11.x) | 2025 年 5 月 7 日 | 2026 年 5 月 7 日 |
|
用于流水线的支持框架
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.15 | 2024 年 4 月 29 日 | 2025 年 4 月 29 日 |
TensorFlow Extended SDK 1.14 | 2023 年 9 月 6 日 | 2024 年 9 月 6 日 |
TensorFlow Extended SDK 1.13 | 2023 年 5 月 3 日 | 2024 年 5 月 3 日 |
TensorFlow Extended SDK 1.12 | 2022 年 12 月 19 日 | 2023 年 12 月 19 日 |
TensorFlow Extended SDK 1.11 | 2022 年 11 月 23 日 | 2023 年 11 月 23 日 |
TensorFlow Extended SDK 1.10 | 2022 年 10 月 4 日 | 2023 年 10 月 4 日 |
TensorFlow Extended SDK 1.9 | 2022 年 7 月 15 日 | 2023 年 7 月 15 日 |
TensorFlow Extended SDK 1.8 | 2022 年 5 月 25 日 | 2023 年 5 月 25 日 |
Google Cloud 流水线组件
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-cu122 (Python 3.10) | CUDA 12.2 | GPU (CUDA 12.2) | 2025 年 1 月 8 日 | 2026 年 1 月 8 日 | common-cu122-VERSION_DATE-debian-11-py310 |
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.17 (Python 3.10) | 2.17.0 | 仅 CPU | 2025 年 7 月 11 日 | 2026 年 7 月 11 日 | tf-2-17-cpu-VERSION_DATE-py310 |
2.17 (Python 3.10) | 2.17.0 | GPU (CUDA 12.3) | 2025 年 7 月 11 日 | 2026 年 7 月 11 日 | tf-2-17-cu123-VERSION_DATE-py310 |
2.16 (Python 3.10) | 2.16.2 | 仅 CPU | 2025 年 6 月 28 日 | 2026 年 6 月 28 日 | tf-2-16-cpu-VERSION_DATE-py310 |
2.16 (Python 3.10) | 2.16.2 | GPU (CUDA 12.3) | 2025 年 6 月 28 日 | 2026 年 6 月 28 日 | tf-2-16-cu123-VERSION_DATE-py310 |
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 | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 | tf-2-6-cpu-VERSION_DATE-py39 |
2.6 (py39) | 2.6.5 | GPU (CUDA 11.3) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 | 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 |