本页面可帮助您选择要使用的容器映像。
选择容器映像类型
Deep Learning Containers 根据时间表支持每个框架版本，以最大限度地减少安全漏洞。查看 Deep Learning Containers 框架支持政策，了解停止提供支持和停止发布日期的影响。
每个容器映像都提供 Python 3 环境，并包含选定的数据科学框架（例如 PyTorch 或 TensorFlow）、Conda、GPU 映像的 NVIDIA 堆栈（CUDA、cuDNN、NCCL2）以及许多其他支持软件包和工具。如需查找适当的容器映像，请参阅下表。
基础版本
| 机器学习框架版本 | 当前补丁程序版本 | 支持的加速器 | 补丁程序和支持日期结束 | 提供终止日期 | 映像系列名称 |
|---|---|---|---|---|---|
| Base-cu121 | CUDA 12.1 (Python 3.10) | GPU (CUDA 12.1) | 2024 年 2 月 28 日 | 2025 年 2 月 28 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310 |
| 基础 CU118 | CUDA 11.8 (Python 3.10) | GPU (CUDA 11.8) | 2024 年 7 月 1 日 | 2025 年 7 月 1 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-gpu.py310 |
| Base-cu113 (Python 3.10) | CUDA 11.3 | GPU (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 | GPU (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 | GPU (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 | GPU (CUDA 11.0) | 2023 年 9 月 1 日 | 2024 年 9 月 1 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37 |
TensorFlow 版本
| 机器学习框架版本 | 当前补丁程序版本 | 支持的加速器 | 补丁程序和支持日期结束 | 提供终止日期 | 映像系列名称 |
|---|---|---|---|---|---|
| 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 | GPU (CUDA 12.1) | 2024 年 11 月 14 日 | 2025 年 11 月 14 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 2024 年 9 月 26 日 | 2025 年 9 月 26 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 2024 年 7 月 5 日 | 2025 年 7 月 5 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 2024 年 6 月 30 日 | 2025 年 6 月 30 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.3) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.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 | GPU (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 | GPU (CUDA 11.3) | 2023 年 11 月 15 日 | 2024 年 11 月 15 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.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 | GPU (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 | GPU (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 | GPU (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 | GPU (CUDA 11.3) | 2023 年 8 月 10 日 | 2024 年 8 月 10 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.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 | GPU (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 | GPU (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-gpu.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-gpu.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-gpu.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-gpu.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-gpu.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 容器映像。
文本生成推断容器
| 机器学习框架版本 | 当前补丁程序版本 | 支持的加速器 | 补丁程序和支持日期结束 | 提供终止日期 | 映像系列名称 |
|---|---|---|---|---|---|
| PyTorch 2.1 (Python 3.10) | 2.1.1 | CUDA 12.1 | 2024 年 10 月 4 日 | 2025 年 10 月 4 日 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310 |
实验性映像系列
此处未列出的映像系列处于实验阶段。我们会尽最大努力支持实验性映像系列，在框架每次发布新版本时，都不会刷新。
列出所有可用的版本
如果您需要特定框架或 CUDA 版本，请搜索可用容器映像的完整列表。如需列出所有可用的 Deep Learning Containers 映像，请在首选终端的 Google Cloud CLI 或 Cloud 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。