En esta página se explica cómo elegir la imagen de contenedor que quieres usar.
Elegir un tipo de imagen de contenedor
Deep Learning Containers admite cada versión del framework según una programación para minimizar las vulnerabilidades de seguridad. Consulta la política de asistencia de frameworks de los contenedores de aprendizaje profundo para conocer las consecuencias de las fechas de finalización de la asistencia y de la disponibilidad.
Cada imagen de contenedor proporciona un entorno de Python 3 e incluye el framework de ciencia de datos seleccionado (como PyTorch o TensorFlow), Conda, la pila de NVIDIA para imágenes de GPU (CUDA, cuDNN, NCCL2) y muchos otros paquetes y herramientas de asistencia. Para encontrar la imagen de contenedor adecuada, consulta las tablas que se incluyen más abajo.
Versiones base
| Versión de framework de aprendizaje automático | Versión de parche actual | Aceleradores admitidos | Fecha de finalización de los parches y la asistencia | Fecha de fin de disponibilidad | Nombre de familia de la imagen |
|---|---|---|---|---|---|
| Base-cu124 | CUDA 12.4 (Python 3.10) | CUDA 12.4 | 1 de abril del 2025 | 1 de abril del 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 enero del 2025 | 1 de enero del 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 enero del 2025 | 8 de enero del 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 feb. 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 | 1 de julio del 2024 | 1 de julio del 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 del 2024 | 1 de enero del 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 sep. 2023 | 1 de septiembre del 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 del 2024 | 1 de enero del 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 sep. 2023 | 1 de septiembre del 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-cu110.py37 |
Versiones de TensorFlow
| Versión de framework de aprendizaje automático | Versión de parche actual | Aceleradores admitidos | Fecha de finalización de los parches y la asistencia | Fecha de fin de disponibilidad | Nombre de familia de la imagen |
|---|---|---|---|---|---|
| 2.17 (Python 3.10) | 2.17.0 | Solo CPU | 11 de julio del 2025 | 11 de julio del 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 del 2025 | 11 de julio del 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 del 2025 | 28 de junio del 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 del 2025 | 28 de junio del 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 del 2024 | 14 nov. 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 del 2024 | 14 nov. 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 | Sep 26, 2024 | 26 de septiembre del 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 | 26 de septiembre del 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 del 2024 | 5 de julio del 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 del 2024 | 5 de julio del 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 del 2024 | 30 de junio del 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 del 2024 | 30 de junio del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 15 de noviembre del 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 del 2023 | 10 de agosto del 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 del 2023 | 10 de agosto del 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 sep. 2023 | 1 de septiembre del 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 sep. 2023 | 1 de septiembre del 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-6.py37 |
| 2.3 | 2.3.4 | Solo CPU | 1 sep. 2023 | 1 de septiembre del 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-cpu.2-3.py37 |
| 2.3 | 2.3.4 | CUDA 11.3 | 1 sep. 2023 | 1 de septiembre del 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.2-3.py37 |
Versiones de PyTorch
| Versión de framework de aprendizaje automático | Versión de parche actual | Aceleradores admitidos | Fecha de finalización de los parches y la asistencia | Fecha de fin de disponibilidad | Nombre de familia de la imagen |
|---|---|---|---|---|---|
| 2.4 (Python 3.10) | 2.4.0 | CUDA 12.4 | 24 de julio del 2025 | 24 de julio del 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 del 2025 | 24 de abril del 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 del 2025 | 30 de enero del 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 del 2024 | 4 oct. 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 mar. 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 | 8 de diciembre del 2023 | 8 dic., 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 del 2023 | 8 dic., 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 del 2023 | 8 dic., 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu113.1-12.py310 |
| 1.12 | 1.12.1 | CUDA 11.3 | 1 sep. 2023 | 1 de septiembre del 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-gpu.1-12.py37 |
Imágenes de contenedor de Model Garden
En esta sección se enumeran las imágenes de contenedor de Model Garden disponibles.
Contenedores de inferencia de vLLM
| Versión del framework | Versión de parche actual | Aceleradores admitidos | Fecha de finalización de los parches y la asistencia | Fecha de fin de disponibilidad | Nombre de familia de la imagen |
|---|---|---|---|---|---|
| PyTorch 2.4 | 2.4.0 | CUDA 12.1 | 21 de agosto del 2025 | 21 de agosto del 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 se admiten en la medida de lo posible y no reciben actualizaciones en cada nueva versión del framework.
Mostrar todas las versiones disponibles
Si necesitas un framework o una versión de CUDA específicos, busca en la lista completa de imágenes de contenedor disponibles. Para ver una lista de todas las imágenes de contenedores de aprendizaje profundo disponibles, usa el siguiente comando en la CLI de Google Cloud con tu terminal preferido o en Cloud Shell.
gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io"
Usar localmente
Los contenedores de aprendizaje profundo se pueden extraer y usar de forma local. Para ello, consulta el artículo Primeros pasos con un contenedor de aprendizaje profundo local.
Ver las ocurrencias de vulnerabilidades
Para ver las repeticiones de vulnerabilidades de una imagen de Deep Learning Containers, consulta Ver repeticiones con gcloud.
Siguientes pasos
- Consulta la información general sobre los contenedores de aprendizaje profundo para obtener más información sobre lo que está preinstalado en las imágenes de contenedor.
- Para empezar a usar los contenedores de aprendizaje profundo, consulta las guías prácticas, que te explican cómo crear y enviar imágenes de contenedores de aprendizaje profundo.