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-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.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 Hugging Face disponibles imágenes de contenedor.
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 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 |
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 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.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 |
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.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/gcr.io/release.huggingface-pytorch-inference-cpu.2-2.transformers.4-41.ubuntu2204.py311 |
Familias de imágenes experimentales
Las familias de imágenes que no se mencionan aquí 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 o una versión de CUDA específicos, 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.