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 minimizar las vulnerabilidades de seguridad. Revisa la política de compatibilidad del framework de los contenedores de aprendizaje profundo 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 tablas que aparecen a continuación.
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-CU121 | CUDA 12.1 (Python 3.10) | GPU (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) | GPU (CUDA 11.8) | 1 de julio de 2024 | 1 de julio de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/base-gpu.py310 |
| Base-cu113 (Python 3.10) | CUDA 11.3 | GPU (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 | GPU (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 | GPU (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 | GPU (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.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 | GPU (CUDA 12.1) | 14 de noviembre de 2024 | 14 de noviembre de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 26 de septiembre de 2024 | 26 de septiembre de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 5 de julio de 2024 | 5 de julio de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (CUDA 11.8) | 30 de junio de 2024 | 30 de junio de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf2-gpu.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 | GPU (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-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 | GPU (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 | GPU (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-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 | GPU (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 | GPU (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 | GPU (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 | GPU (CUDA 11.3) | 10 de agosto de 2023 | 10 de agosto de 2024 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/tf-gpu.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 | GPU (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 | GPU (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-gpu.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-gpu.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-gpu.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-gpu.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-gpu.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 cara abrazada
En esta sección, se enumeran las imágenes de contenedor disponibles de Hugging Face.
Contenedores de inferencia de generación de texto
| 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 |
|---|---|---|---|---|---|
| PyTorch 2.1 (Python 3.10) | 2.1.1 | CUDA 12.1 | 4 de octubre de 2024 | 4 de octubre de 2025 | us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu121.1-4.ubuntu2204.py310 |
Familias de imágenes experimentales
Las familias de imágenes que no se incluyen en esta lista son experimentales. Las familias de imágenes experimentales se admiten 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 en 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.