選擇容器映像檔

本頁面可協助您選擇要使用的容器映像檔。

選擇容器映像檔類型

深度學習容器會依據排程支援各架構版本，盡量減少安全漏洞。請參閱 Deep Learning Containers 架構支援政策，瞭解終止支援和終止供應日期帶來的影響。

每個容器映像檔都提供 Python 3 環境，並包含所選的資料科學架構 (例如 PyTorch 或 TensorFlow)、Conda、GPU 映像檔的 NVIDIA 堆疊 (CUDA、cuDNN、NCCL2)，以及許多其他支援套件和工具。如要尋找合適的容器映像檔，請參閱下表。

基礎版本

ML 架構版本 目前的修補程式版本 支援的加速器 修補程式和支援服務終止日期 供應終止日期 映像檔系列名稱
Base-cu124 CUDA 12.4 (Python 3.10) CUDA 12.4 2025 年 4 月 1 日 2026 年 4 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu124.py310
Base-cu123 CUDA 12.3 (Python 3.10) CUDA 12.3 2025 年 1 月 1 日 2026 年 1 月 1 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu123.py310
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 版本

ML 架構版本 目前的修補程式版本 支援的加速器 修補程式和支援服務終止日期 供應終止日期 映像檔系列名稱
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 Nov 14, 2024 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 Nov 14, 2024 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 版本

ML 架構版本 目前的修補程式版本 支援的加速器 修補程式和支援服務終止日期 供應終止日期 映像檔系列名稱
2.4 (Python 3.10) 2.4.0 CUDA 12.4 2025 年 7 月 24 日 2026 年 7 月 24 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu124.2-4.py310
2.3 (Python 3.10) 2.3.0 CUDA 12.1 2025 年 4 月 24 日 2026 年 4 月 24 日 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu121.2-3.py310
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 Oct 4, 2024 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

模型園地容器映像檔

本節列出可用的 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 CLI 中使用下列指令，或在偏好的終端機或 Cloud Shell 中使用下列指令。

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

在本機使用

您可以提取深度學習容器，並在本機使用。如要瞭解如何操作，請參閱「開始使用本機深度學習容器」。

查看安全漏洞例項

如要查看 Deep Learning Containers 映像檔的安全漏洞例項，請參閱「使用 gcloud 查看例項」。

後續步驟

  • 請參閱深度學習容器總覽，進一步瞭解容器映像檔預先安裝的項目。
  • 如要開始使用深度學習容器，請參閱使用指南，瞭解如何建構及推送深度學習容器映像檔。