Choose a container image

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.