This page helps you choose which container image to use.
Choose a container image type
Deep Learning Containers supports each framework version based on a schedule to minimize security vulnerabilities. Review the Deep Learning Containers framework support policy to understand the implications of the end-of-support and end-of-availability dates.
Each container image provides a Python 3 environment and includes the selected data science framework (such as PyTorch or TensorFlow), Conda, the NVIDIA stack for GPU images (CUDA, cuDNN, NCCL2), and many other supporting packages and tools. To find the appropriate container image, see the tables below.
Base versions
ML framework version | Current patch version | Supported accelerators | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|
Base-cu122 | CUDA 12.2 (Python 3.10) | CUDA 12.2 | Jan 8, 2025 | Jan 8, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu122.py310 |
Base-cu121 | CUDA 12.1 (Python 3.10) | CUDA 12.1 | Feb 28, 2024 | Feb 28, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310 |
Base-cu118 | CUDA 11.8 (Python 3.10) | CUDA 11.8 | Jul 1, 2024 | Jul 1, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu118.py310 |
Base-cu113 (Python 3.10) | CUDA 11.3 | CUDA 11.3 | Jan 1, 2024 | Jan 1, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py310 |
Base-cu113 (Python 3.7) | CUDA 11.3 | CUDA 11.3 | Sep 1, 2023 | Sep 1, 2024 | 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 | Jan 1, 2024 | Jan 1, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py310 |
Base-cu110 (Python 3.7) | CUDA 11.0 | CUDA 11.0 | Sep 1, 2023 | Sep 1, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37 |
TensorFlow versions
ML framework version | Current patch version | Supported accelerators | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|
2.17 (Python 3.10) | 2.17.0 | CPU only | Jul 11, 2025 | Jul 11, 2026 | 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 | Jul 11, 2025 | Jul 11, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-17.py310 |
2.16 (Python 3.10) | 2.16.2 | CPU only | Jun 28, 2025 | Jun 28, 2026 | 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 | Jun 28, 2025 | Jun 28, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-16.py310 |
2.15 (Python 3.10) | 2.15.0 | CPU only | Nov 14, 2024 | Nov 14, 2025 | 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 | Nov 14, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu121.2-15.py310 |
2.14 (Python 3.10) | 2.14.0 | CPU only | Sep 26, 2024 | Sep 26, 2025 | 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 | Sep 26, 2024 | Sep 26, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-14.py310 |
2.13 (Python 3.10) | 2.13.0 | CPU only | Jul 5, 2024 | Jul 5, 2025 | 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 | Jul 5, 2024 | Jul 5, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-13.py310 |
2.12 (Python 3.10) | 2.12.0 | CPU only | Jun 30, 2024 | Jun 30, 2025 | 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 | Jun 30, 2024 | Jun 30, 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-12.py310 |
2.11 (Python 3.10) | 2.11.0 | CPU only | Nov 15, 2023 | Nov 15, 2024 | 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 | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-11.py310 |
2.11 | 2.11.0 | CPU only | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py37 |
2.11 | 2.11.0 | CUDA 11.3 | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-11.py37 |
2.10 (Python 3.10) | 2.10.1 | CPU only | Nov 15, 2023 | Nov 15, 2024 | 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 | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-10.py310 |
2.10 | 2.10.1 | CPU only | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py37 |
2.10 | 2.10.1 | CUDA 11.3 | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-10.py37 |
2.9 | 2.9.3 | CPU only | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-9.py37 |
2.9 | 2.9.3 | CUDA 11.3 | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-9.py37 |
2.8 | 2.8.4 | CPU only | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-8.py37 |
2.8 | 2.8.4 | CUDA 11.3 | Nov 15, 2023 | Nov 15, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-8.py37 |
2.6 (Python 3.9) | 2.6.5 | CPU only | Aug 10, 2023 | Aug 10, 2024 | 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 | Aug 10, 2023 | Aug 10, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-6.py39 |
2.6 (Python 3.7) | 2.6.5 | CPU only | Sep 1, 2023 | Sep 1, 2024 | 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 | Sep 1, 2023 | Sep 1, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37 |
2.3 | 2.3.4 | CPU only | Sep 1, 2023 | Sep 1, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-3.py37 |
2.3 | 2.3.4 | CUDA 11.3 | Sep 1, 2023 | Sep 1, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37 |
PyTorch versions
ML framework version | Current patch version | Supported accelerators | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|
2.3 (Python 3.10) | 2.3.0 | CUDA 12.1 | Apr 24, 2025 | Apr 24, 2026 | 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 | Jan 30, 2025 | Jan 30, 2026 | 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 | Oct 4, 2025 | 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 | Mar 15, 2024 | Mar 15, 2025 | 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 | Dec 8, 2023 | Dec 8, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu118.1-13.py310 |
1.13 | 1.13.1 | CUDA 11.8 | Dec 8, 2023 | Dec 8, 2024 | 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 | Dec 8, 2023 | Dec 8, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu113.1-12.py310 |
1.12 | 1.12.1 | CUDA 11.3 | Sep 1, 2023 | Sep 1, 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37 |
Hugging Face container images
This section lists the available Hugging Face Deep Learning Containers.
Text Generation Inference containers
Framework version | Current patch version | Supported accelerators | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|
TGI 2.3 | 2.3.1 | CUDA 12.4 | Oct 9, 2025 | Oct 9, 2026 | 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 | July 23, 2025 | July 23, 2026 | 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 | June 27, 2025 | June 27, 2026 | 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 | April 25, 2025 | April 25, 2026 | 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 | March 13, 2025 | March 13, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310 |
Text Embeddings Inference containers
Framework version | Current patch version | Supported accelerators | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|
TEI 1.4 | 1.4.0 | CUDA 12.2 | Aug 27, 2025 | Aug 27, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-4.ubuntu2204 |
TEI 1.4 | 1.4.0 | CPU | Aug 27, 2025 | Aug 27, 2026 | 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 | May 2, 2025 | May 2, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-2.ubuntu2204 |
TEI 1.2 | 1.2.2 | CPU | May 2, 2025 | May 2, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-2 |
PyTorch Inference containers
Framework version | Current patch version | Supported accelerators | Transformers version | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|---|
PyTorch 2.2 | 2.2.2 | CUDA 12.1 | 4.41.1 | June 12, 2025 | June 12, 2026 | 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 | June 7, 2025 | June 7, 2026 | 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 | Aug 21, 2025 | Aug 21, 2026 | 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 | Aug 21, 2025 | Aug 21, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311 |
PyTorch training containers
Framework version | Current patch version | Supported accelerators | Transformers version | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|---|
PyTorch 2.3 | 2.3.0 | CUDA 12.1 | 4.42.3 | Aug 21, 2025 | Aug 21, 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.2-3.transformers.4-42.ubuntu2204.py310 |
Model Garden container images
This section lists the available Model Garden container images.
vLLM inference containers
Framework version | Current patch version | Supported accelerators | End of patch and support date | End of availability date | Image family name |
---|---|---|---|---|---|
PyTorch 2.4 | 2.4.0 | CUDA 12.1 | Aug 21, 2025 | Aug 21, 2026 | us-docker.pkg.dev/deeplearning-platform-release/vertex-model-garden/vllm-inference.cu121.0-5.ubuntu2204.py310 |
Experimental image families
Image families that aren't listed here are experimental. Experimental image families are supported on a best-effort basis, and don't receive refreshes on every new release of the framework.
Listing all available versions
If you need a specific framework or CUDA version, search the complete list of available container images. To list all available Deep Learning Containers images, use the following command in the Google Cloud CLI with your preferred terminal or in Cloud Shell.
gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"
Using locally
Deep Learning Containers can be pulled and used locally. To do so, see Getting started with a local deep learning container.
What's next
- Read the Deep Learning Containers overview to learn more about what is pre-installed on container images.
- Get started with Deep Learning Containers by walking through the How-to guides, which provide instructions on how to build and push deep learning container images.