选择容器映像

本页面可帮助您选择要使用的容器映像。

选择容器映像类型

Deep Learning Containers 根据时间表支持每个框架版本, 最大限度地降低安全漏洞。查看 Deep Learning Containers 框架支持政策,了解停止提供支持和停止发布日期的影响。

每个容器映像都提供 Python 3 环境,并包含选定的数据科学框架(例如 PyTorch 或 TensorFlow)、Conda、GPU 映像的 NVIDIA 堆栈(CUDA、cuDNN、NCCL2)以及许多其他支持软件包和工具。如需查找适当的容器映像,请参阅下表。

基础版本

机器学习框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
Base-cu122 CUDA 12.2(Python 3.10) CUDA 12.2 2025 年 1 月 8 日 2026 年 1 月 8 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu122.py310
Base-cu121 CUDA 12.1 (Python 3.10) CUDA 12.1 2024 年 2 月 28 日 2025 年 2 月 28 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310
Base-cu118 CUDA 11.8(Python 3.10) CUDA 11.8 2024 年 7 月 1 日 2025 年 7 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu118.py310
Base-cu113 (Python 3.10) CUDA 11.3 CUDA 11.3 2024 年 1 月 1 日 2025 年 1 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py310
Base-cu113 (Python 3.7) CUDA 11.3 CUDA 11.3 2023 年 9 月 1 日 2024 年 9 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py37
Base-cu110 (Python 3.10 / Debian 11) CUDA 11.0 CUDA 11.0 2024 年 1 月 1 日 2025 年 1 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py310
Base-cu110 (Python 3.7) CUDA 11.0 CUDA 11.0 2023 年 9 月 1 日 2024 年 9 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37

TensorFlow 版本

机器学习框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
2.17 (Python 3.10) 2.17.0 仅 CPU 2025 年 7 月 11 日 2026 年 7 月 11 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-17.py310
2.17(Python 3.10) 2.17.0 CUDA 12.3 2025 年 7 月 11 日 2026 年 7 月 11 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-17.py310
2.16 (Python 3.10) 2.16.2 仅 CPU 2025 年 6 月 28 日 2026 年 6 月 28 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-16.py310
2.16(Python 3.10) 2.16.2 CUDA 12.3 2025 年 6 月 28 日 2026 年 6 月 28 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-16.py310
2.15 (Python 3.10) 2.15.0 仅 CPU 2024 年 11 月 14 日 2025 年 11 月 14 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-15.py310
2.15 (Python 3.10) 2.15.0 CUDA 12.1 2024 年 11 月 14 日 2025 年 11 月 14 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu121.2-15.py310
2.14 (Python 3.10) 2.14.0 仅 CPU 2024 年 9 月 26 日 2025 年 9 月 26 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-14.py310
2.14 (Python 3.10) 2.14.0 CUDA 11.8 2024 年 9 月 26 日 2025 年 9 月 26 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-14.py310
2.13 (Python 3.10) 2.13.0 仅 CPU 2024 年 7 月 5 日 2025 年 7 月 5 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-13.py310
2.13 (Python 3.10) 2.13.0 CUDA 11.8 2024 年 7 月 5 日 2025 年 7 月 5 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-13.py310
2.12 (Python 3.10) 2.12.0 仅 CPU 2024 年 6 月 30 日 2025 年 6 月 30 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-12.py310
2.12 (Python 3.10) 2.12.0 CUDA 11.8 2024 年 6 月 30 日 2025 年 6 月 30 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-12.py310
2.11 (Python 3.10) 2.11.0 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py310
2.11 (Python 3.10) 2.11.0 CUDA 11.3 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-11.py310
2.11 2.11.0 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py37
2.11 2.11.0 CUDA 11.3 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-11.py37
2.10 (Python 3.10) 2.10.1 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py310
2.10 (Python 3.10) 2.10.1 CUDA 11.3 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-10.py310
2.10 2.10.1 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py37
2.10 2.10.1 CUDA 11.3 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-10.py37
2.9 2.9.3 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-9.py37
2.9 2.9.3 CUDA 11.3 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-9.py37
2.8 2.8.4 仅 CPU 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-8.py37
2.8 2.8.4 CUDA 11.3 2023 年 11 月 15 日 2024 年 11 月 15 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-8.py37
2.6 (Python 3.9) 2.6.5 仅 CPU 2023 年 8 月 10 日 2024 年 8 月 10 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-6.py39
2.6 (Python 3.9) 2.6.5 CUDA 11.3 2023 年 8 月 10 日 2024 年 8 月 10 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-6.py39
2.6 (Python 3.7) 2.6.5 仅 CPU 2023 年 9 月 1 日 2024 年 9 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-6.py37
2.6 (Python 3.7) 2.6.5 CUDA 11.3 2023 年 9 月 1 日 2024 年 9 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37
2.3 2.3.4 仅 CPU 2023 年 9 月 1 日 2024 年 9 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-3.py37
2.3 2.3.4 CUDA 11.3 2023 年 9 月 1 日 2024 年 9 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37

PyTorch 版本

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

Hugging Face 容器映像

本部分列出了可用的 Hugging Face Deep Learning Containers

文本生成推理容器

框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
TGI 2.3 2.3.1 CUDA 12.4 2025 年 10 月 9 日 2026 年 10 月 9 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu124.2-3.ubuntu2204.py311
TGI 2.2 2.2.0 CUDA 12.1 2025 年 7 月 23 日 2026 年 7 月 23 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-2.ubuntu2204.py310
TGI 2.1 2.1.1 CUDA 12.1 2025 年 6 月 27 日 2026 年 6 月 27 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-1.ubuntu2204.py310
TGI 2.0 2.0.1 CUDA 12.1 2025 年 4 月 25 日 2026 年 4 月 25 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.2-0.ubuntu2204.py310
TGI 1.4 1.4.4 CUDA 12.1 2025 年 3 月 13 日 2026 年 3 月 13 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310

文本嵌入推理容器

框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
TEI 1.4 1.4.0 CUDA 12.2 2025 年 8 月 27 日 2026 年 8 月 27 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-4.ubuntu2204
TEI 1.4 1.4.0 CPU 2025 年 8 月 27 日 2026 年 8 月 27 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-4
TEI 1.2 1.2.2 CUDA 12.2 2025 年 5 月 2 日 2026 年 5 月 2 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-2.ubuntu2204
TEI 1.2 1.2.2 CPU 2025 年 5 月 2 日 2026 年 5 月 2 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-2

PyTorch 推断容器

框架版本 当前补丁程序版本 支持的加速器 Transformer 版本 补丁程序和支持日期结束 提供终止日期 映像系列名称
PyTorch 2.2 2.2.2 CUDA 12.1 4.41.1 2025 年 6 月 12 日 2026 年 6 月 12 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-41.ubuntu2204.py311
PyTorch 2.2 2.2.2 CPU 4.41.1 2025 年 6 月 7 日 2026 年 6 月 7 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-41.ubuntu2204.py311
PyTorch 2.2 2.2.2 CUDA 12.1 4.44.0 2025 年 8 月 21 日 2026 年 8 月 21 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-44.ubuntu2204.py311
PyTorch 2.2 2.2.2 CPU 4.44.0 2025 年 8 月 21 日 2026 年 8 月 21 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311

PyTorch 训练容器

框架版本 当前补丁程序版本 支持的加速器 Transformer 版本 补丁程序和支持日期结束 提供终止日期 映像系列名称
PyTorch 2.3 2.3.0 CUDA 12.1 4.42.3 2025 年 8 月 21 日 2026 年 8 月 21 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.2-3.transformers.4-42.ubuntu2204.py310

Model Garden 容器映像

本部分列出了可用的 Model Garden 容器映像。

vLLM 推理容器

框架版本 当前补丁程序版本 支持的加速器 补丁程序和支持日期结束 提供终止日期 映像系列名称
PyTorch 2.4 2.4.0 CUDA 12.1 2025 年 8 月 21 日 2026 年 8 月 21 日 us-docker.pkg.dev/deeplearning-platform-release/vertex-model-garden/vllm-inference.cu121.0-5.ubuntu2204.py310

实验性映像系列

此处未列出的映像系列处于实验阶段。实验性映像系列受到了最大程度的支持,但在框架每次发布新版本时不会刷新。

列出所有可用的版本

如果您需要特定框架或 CUDA 版本,请搜索可用容器映像的完整列表。如需列出所有可用的 Deep Learning Containers 映像,请在首选终端的 Google Cloud CLICloud Shell 中使用以下命令。

gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"

在本地使用

您可以在本地拉取和使用 Deep Learning Containers。为此,请参阅本地 Deep Learning Containers 使用入门

后续步骤

  • 阅读 Deep Learning Containers 概览,详细了解容器映像中预安装的内容。
  • 逐步浏览方法指南,了解如何构建和推送深度学习容器映像,以开始使用 Deep Learning Containers。