En esta página, encontrarás ayuda para elegir qué imagen de contenedor usar.
Elige un tipo de imagen de contenedor
Contenedores de aprendizaje profundo admite cada versión de framework según un programa para minimizar las vulnerabilidades de seguridad. Revisa la política de asistencia del framework de Deep Learning Containers para comprender las implicaciones de las fechas de finalización de la asistencia y de finalización de la disponibilidad.
Cada imagen de contenedor proporciona un entorno de Python 3 y, además, incluye el marco de trabajo de ciencia de datos seleccionado (como PyTorch o TensorFlow), Conda, la pila de NVIDIA para las imágenes de GPU (CUDA, cuDNN, NCCL2) y muchos otros paquetes de asistencia y herramientas. Para encontrar la imagen de contenedor adecuada, consulta las siguientes tablas.
Versiones de base
| Versión del marco de trabajo de AA | Versión actual del parche | Aceleradores compatibles | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|
| Base-cu122 | CUDA 12.2 (Python 3.10) | CUDA 12.2 | 8 de enero de 2025 | 8 de enero 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 febrero de 2024 | 28 de febrero 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 julio de 2024 | 1 de julio 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 enero de 2024 | 1 de enero 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 septiembre de 2023 | 1 de septiembre 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 enero de 2024 | 1 de enero 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 septiembre de 2023 | 1 de septiembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37 |
Versiones de TensorFlow
| Versión del marco de trabajo de AA | Versión actual del parche | Aceleradores compatibles | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|
| 2.17 (Python 3.10) | 2.17.0 | Solo CPU | 11 de julio de 2025 | 11 de julio 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 julio de 2025 | 11 de julio de 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-17.py310 |
| 2.16 (Python 3.10) | 2.16.2 | Solo CPU | 28 de junio de 2025 | 28 de junio 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 junio de 2025 | 28 de junio de 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu123.2-16.py310 |
| 2.15 (Python 3.10) | 2.15.0 | Solo CPU | 14 de noviembre de 2024 | 14 de noviembre 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 noviembre de 2024 | 14 de noviembre de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu121.2-15.py310 |
| 2.14 (Python 3.10) | 2.14.0 | Solo CPU | 26 de septiembre de 2024 | 26 de septiembre 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 septiembre de 2024 | 26 de septiembre de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-14.py310 |
| 2.13 (Python 3.10) | 2.13.0 | Solo CPU | 5 de julio de 2024 | 5 de julio 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 julio de 2024 | 5 de julio de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-13.py310 |
| 2.12 (Python 3.10) | 2.12.0 | Solo CPU | 30 de junio de 2024 | 30 de junio 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 junio de 2024 | 30 de junio de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-cu118.2-12.py310 |
| 2.11 (Python 3.10) | 2.11.0 | Solo CPU | 15 de noviembre de 2023 | 15 de noviembre 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 noviembre de 2023 | 15 de noviembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-11.py310 |
| 2.11 | 2.11.0 | Solo CPU | 15 de noviembre de 2023 | 15 de noviembre 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 noviembre de 2023 | 15 de noviembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-11.py37 |
| 2.10 (Python 3.10) | 2.10.1 | Solo CPU | 15 de noviembre de 2023 | 15 de noviembre 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 noviembre de 2023 | 15 de noviembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cu113.2-10.py310 |
| 2.10 | 2.10.1 | Solo CPU | 15 de noviembre de 2023 | 15 de noviembre 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 noviembre de 2023 | 15 de noviembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-10.py37 |
| 2.9 | 2.9.3 | Solo CPU | 15 de noviembre de 2023 | 15 de noviembre 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 noviembre de 2023 | 15 de noviembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-9.py37 |
| 2.8 | 2.8.4 | Solo CPU | 15 de noviembre de 2023 | 15 de noviembre 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 noviembre de 2023 | 15 de noviembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.2-8.py37 |
| 2.6 (Python 3.9) | 2.6.5 | Solo 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 | Solo CPU | 1 de septiembre de 2023 | 1 de septiembre 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 septiembre de 2023 | 1 de septiembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37 |
| 2.3 | 2.3.4 | Solo CPU | 1 de septiembre de 2023 | 1 de septiembre 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 septiembre de 2023 | 1 de septiembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37 |
Versiones de PyTorch
| Versión del marco de trabajo de AA | Versión actual del parche | Aceleradores compatibles | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|
| 2.3 (Python 3.10) | 2.3.0 | CUDA 12.1 | 24 de abril, 2025 | 24 de abril, 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 enero de 2025 | 30 de enero 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 octubre de 2024 | 4 de octubre 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 marzo de 2024 | 15 de marzo 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 diciembre de 2023 | 8 de diciembre de 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 diciembre de 2023 | 8 de diciembre de 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 diciembre de 2023 | 8 de diciembre de 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 septiembre de 2023 | 1 de septiembre de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37 |
Imágenes de contenedor de Hugging Face
En esta sección, se enumeran los contenedores de aprendizaje profundo de Hugging Face disponibles.
Contenedores de inferencia de generación de texto
| Versión del framework | Versión actual del parche | Aceleradores compatibles | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|
| TGI 2.3 | 2.3.1 | CUDA 12.4 | 9 de octubre de 2025 | 9 de octubre de 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 | 23 de julio de 2025 | 23 de julio de 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 | 27 de junio de 2025 | 27 de junio de 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 | 25 de abril de 2025 | 25 de abril de 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 | 13 de marzo de 2025 | 13 de marzo de 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310 |
Contenedores de inferencia de incorporaciones de texto
| Versión del framework | Versión actual del parche | Aceleradores compatibles | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|
| TEI 1.4 | 1.4.0 | CUDA 12.2 | 27 de agosto de 2025 | 27 de agosto de 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 | 27 de agosto de 2025 | 27 de agosto de 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 | 2 de mayo de 2025 | 2 de mayo de 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 | 2 de mayo de 2025 | 2 de mayo de 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-2 |
Contenedores de inferencia de PyTorch
| Versión del framework | Versión actual del parche | Aceleradores compatibles | Versión de Transformers | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|---|
| PyTorch 2.2 | 2.2.2 | CUDA 12.1 | 4.41.1 | 12 de junio de 2025 | 12 de junio de 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 | 7 de junio de 2025 | 7 de junio de 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 | 21 de agosto de 2025 | 21 de agosto de 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 | 21 de agosto de 2025 | 21 de agosto de 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311 |
Contenedores de entrenamiento de PyTorch
| Versión del framework | Versión actual del parche | Aceleradores compatibles | Versión de Transformers | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|---|
| PyTorch 2.3 | 2.3.0 | CUDA 12.1 | 4.42.3 | 21 de agosto de 2025 | 21 de agosto de 2026 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.2-3.transformers.4-42.ubuntu2204.py310 |
Imágenes de contenedores de Model Garden
En esta sección, se enumeran las imágenes de contenedores disponibles de Model Garden.
Contenedores de inferencia de vLLM
| Versión del framework | Versión actual del parche | Aceleradores compatibles | Fecha de finalización del parche y la compatibilidad | Fecha de finalización de la disponibilidad | Nombre de la familia de imágenes |
|---|---|---|---|---|---|
| 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 |
Familias de imágenes experimentales
Las familias de imágenes que no aparecen en esta lista son experimentales. Las familias de imágenes experimentales son compatibles en función del mejor esfuerzo y no reciben actualizaciones en cada versión nueva del framework.
Muestra todas las versiones disponibles
Si necesitas un framework específico o una versión de CUDA, busca la lista completa de imágenes de contenedor disponibles. Para enumerar todas las imágenes de contenedores de aprendizaje profundo disponibles, usa el siguiente comando en Google Cloud CLI con tu terminal preferida o en Cloud Shell.
gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"
Uso local
Los contenedores de aprendizaje profundo se pueden extraer y usar de forma local. Para ello, consulta Comienza a usar un contenedor de aprendizaje profundo local.
¿Qué sigue?
- Lee la descripción general de los contenedores de aprendizaje profundo para obtener más información sobre los elementos preinstalados en las imágenes de contenedor.
- Comienza con los contenedores de aprendizaje profundo mediante las guías prácticas, que proporcionan instrucciones para compilar y enviar imágenes de contenedores de aprendizaje profundo.