Escolha uma imagem do contentor

Esta página ajuda a escolher a imagem do contentor a usar.

Escolha um tipo de imagem do contentor

Os Deep Learning Containers suportam cada versão da framework com base num horário para minimizar as vulnerabilidades de segurança. Reveja a Política de apoio técnico da framework Deep Learning Containers para compreender as implicações das datas de fim do apoio técnico e fim da disponibilidade.

Cada imagem de contentor fornece um ambiente Python 3 e inclui a framework de ciência de dados selecionada (como PyTorch ou TensorFlow), o Conda, a pilha NVIDIA para imagens de GPU (CUDA, cuDNN, NCCL2) e muitos outros pacotes e ferramentas de apoio. Para encontrar a imagem do contentor adequada, consulte as tabelas abaixo.

Versões base

Versão do framework de ML Versão de patch atual Aceleradores suportados Data de fim da aplicação de patches e do apoio técnico Data de fim da disponibilidade Nome de família da imagem
Base-cu124 CUDA 12.4 (Python 3.10) CUDA 12.4 1 de abril de 2025 1 de abril de 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu124.py310
Base-cu123 CUDA 12.3 (Python 3.10) CUDA 12.3 1 de janeiro de 2025 1 de janeiro de 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu123.py310
Base-cu122 CUDA 12.2 (Python 3.10) CUDA 12.2 8 de janeiro de 2025 8 de janeiro de 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu122.py310
Base-cu121 CUDA 12.1 (Python 3.10) CUDA 12.1 28 de fevereiro de 2024 28 de fevereiro de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu121.py310
Base-cu118 CUDA 11.8 (Python 3.10) CUDA 11.8 1 de julho de 2024 1 de julho de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu118.py310
Base-cu113 (Python 3.10) CUDA 11.3 CUDA 11.3 1 de janeiro de 2024 1 de janeiro de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu113.py310
Base-cu113 (Python 3.7) CUDA 11.3 CUDA 11.3 1 de setembro de 2023 1 de setembro de 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 1 de janeiro de 2024 1 de janeiro de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py310
Base-cu110 (Python 3.7) CUDA 11.0 CUDA 11.0 1 de setembro de 2023 1 de setembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37

Versões do TensorFlow

Versão do framework de ML Versão de patch atual Aceleradores suportados Data de fim da aplicação de patches e do apoio técnico Data de fim da disponibilidade Nome de família da imagem
2.17 (Python 3.10) 2.17.0 Apenas CPU 11 de julho de 2025 11 de julho de 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 11 de julho de 2025 11 de julho de 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-17.py310
2.16 (Python 3.10) 2.16.2 Apenas CPU 28 de junho de 2025 28 de junho de 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 28 de junho de 2025 28 de junho de 2026 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-16.py310
2.15 (Python 3.10) 2.15.0 Apenas CPU 14 de novembro de 2024 14 de novembro de 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 14 de novembro de 2024 14 de novembro de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu121.2-15.py310
2.14 (Python 3.10) 2.14.0 Apenas CPU 26 de setembro de 2024 26 de setembro de 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 26 de setembro de 2024 26 de setembro de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-14.py310
2.13 (Python 3.10) 2.13.0 Apenas CPU 5 de julho de 2024 5 de julho de 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 5 de julho de 2024 5 de julho de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-13.py310
2.12 (Python 3.10) 2.12.0 Apenas CPU 30 de junho de 2024 30 de junho de 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 30 de junho de 2024 30 de junho de 2025 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-12.py310
2.11 (Python 3.10) 2.11.0 Apenas CPU 15 de novembro de 2023 15 de novembro de 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 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-11.py310
2.11 2.11.0 Apenas CPU 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-11.py37
2.11 2.11.0 CUDA 11.3 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-11.py37
2.10 (Python 3.10) 2.10.1 Apenas CPU 15 de novembro de 2023 15 de novembro de 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 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-10.py310
2.10 2.10.1 Apenas CPU 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-10.py37
2.10 2.10.1 CUDA 11.3 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-10.py37
2.9 2.9.3 Apenas CPU 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-9.py37
2.9 2.9.3 CUDA 11.3 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-9.py37
2.8 2.8.4 Apenas CPU 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cpu.2-8.py37
2.8 2.8.4 CUDA 11.3 15 de novembro de 2023 15 de novembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-8.py37
2.6 (Python 3.9) 2.6.5 Apenas CPU 10 de agosto de 2023 10 de agosto de 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 10 de agosto de 2023 10 de agosto de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-6.py39
2.6 (Python 3.7) 2.6.5 Apenas CPU 1 de setembro de 2023 1 de setembro de 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 1 de setembro de 2023 1 de setembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37
2.3 2.3.4 Apenas CPU 1 de setembro de 2023 1 de setembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-3.py37
2.3 2.3.4 CUDA 11.3 1 de setembro de 2023 1 de setembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37

Versões do PyTorch

Versão do framework de ML Versão de patch atual Aceleradores suportados Data de fim da aplicação de patches e do apoio técnico Data de fim da disponibilidade Nome de família da imagem
2.4 (Python 3.10) 2.4.0 CUDA 12.4 24 de julho de 2025 24 de julho de 2026 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 24 de abril de 2025 24 de abril de 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 30 de janeiro de 2025 30 de janeiro de 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 4 de outubro de 2024 4 de outubro de 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 15 de março de 2024 15 de março de 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 8 de dezembro de 2023 8 de dez. 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu118.1-13.py310
1.13 1.13.1 CUDA 11.8 8 de dezembro de 2023 8 de dez. 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 8 de dezembro de 2023 8 de dez. 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu113.1-12.py310
1.12 1.12.1 CUDA 11.3 1 de setembro de 2023 1 de setembro de 2024 us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37

Imagens de contentores do Model Garden

Esta secção apresenta as imagens de contentores do Model Garden disponíveis.

Contentores de inferência de vLLM

Versão da framework Versão de patch atual Aceleradores suportados Data de fim da aplicação de patches e do apoio técnico Data de fim da disponibilidade Nome de família da imagem
PyTorch 2.4 2.4.0 CUDA 12.1 21 de agosto de 2025 21 de agosto de 2026 us-docker.pkg.dev/deeplearning-platform-release/vertex-model-garden/vllm-inference.cu121.0-5.ubuntu2204.py310

Famílias de imagens experimentais

As famílias de imagens que não estão listadas aqui são experimentais. As famílias de imagens experimentais são suportadas com base no melhor esforço possível e não recebem atualizações em cada novo lançamento da framework.

Indicar todas as versões disponíveis

Se precisar de uma framework ou uma versão do CUDA específica, pesquise a lista completa de imagens de contentores disponíveis. Para listar todas as imagens dos Deep Learning Containers disponíveis, use o seguinte comando na CLI Google Cloud com o seu terminal preferencial ou na Cloud Shell.

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

Usar localmente

Os Deep Learning Containers podem ser extraídos e usados localmente. Para tal, consulte o artigo Começar a usar um contentor de aprendizagem profunda local.

Veja ocorrências de vulnerabilidades

Para ver as ocorrências de vulnerabilidades de uma imagem do Deep Learning Containers, consulte Ver ocorrências através do gcloud.

O que se segue?