Elegir una imagen de contenedor

En esta página, encontrarás ayuda para elegir qué imagen de contenedor usar.

Elige un tipo de imagen de contenedor

Los contenedores de aprendizaje profundo admiten cada versión del framework según un programa para y minimizar las vulnerabilidades de seguridad. Revisa el framework de los contenedores de aprendizaje profundo sobre la política de asistencia al cliente para comprender el 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 ene de 2025 8 de ene 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.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 del 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 los Model Garden disponibles imágenes de contenedor.

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. Imagen experimental familias reciben la mejor asistencia posible y no reciben actualizaciones sobre con 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?