支持的框架

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

如需深入了解框架支持政策,请参阅Agent Platform 框架支持政策。

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

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

TensorFlow

机器学习框架版本 支持的加速器(以及 CUDA 版本,如适用) 补丁程序和支持日期结束 提供终止日期 支持的图片
2.17 (Python 3.10) 仅 CPU 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
2.17 (Python 3.10) GPU (CUDA 12.x) 2025 年 7 月 11 日 2026 年 7 月 11 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
2.16 (Python 3.10) 仅 CPU 2025 年 6 月 28 日 2026 年 6 月 28 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
2.16 (Python 3.10) GPU (CUDA 12.x) 2025 年 6 月 28 日 2026 年 6 月 28 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
2.15 (Python 3.10) 仅 CPU 2024 年 11 月 14 日 2025 年 11 月 14 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
2.15 (Python 3.10) GPU (CUDA 12.x) 2024 年 11 月 14 日 2025 年 11 月 14 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
2.15 (Python 3.10) 仅限 TPU 2024 年 10 月 23 日 2025 年 10 月 23 日
  • us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest
2.14 (Python 3.10) 仅 CPU 2024 年 9 月 26 日 2025 年 9 月 26 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest
2.14 (Python 3.10) GPU (CUDA 11.x) 2024 年 9 月 26 日 2025 年 9 月 26 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest
2.13 (Python 3.10) 仅 CPU 2024 年 7 月 5 日 2025 年 7 月 5 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest
2.13 (Python 3.10) GPU (CUDA 11.x) 2024 年 7 月 5 日 2025 年 7 月 5 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest
2.12 (Python 3.10) 仅 CPU 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest
2.12 (Python 3.10) GPU (CUDA 11.x) 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest
2.12 (Python 3.8) 仅限 TPU 2024 年 6 月 30 日 2025 年 6 月 30 日
  • us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest
2.11 (Python 3.10) 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest
2.11 (Python 3.10) GPU (CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest
2.11 (Python 3.7) 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest
2.11 (Python 3.7) GPU (CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest
2.9 (Python 3.10) 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest
2.9 (Python 3.10) GPU (CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest
2.9 (Python 3.7) 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest
2.9 (Python 3.7) GPU (CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest
2.8 (Python 3.10) 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest
2.8 (Python 3.10) GPU (CUDA 11.x) 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest
2.8 (Python 3.7) 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest