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

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. A fin de buscar la imagen de contenedor que deseas, consulta la siguiente tabla.

La siguiente lista de tipos de imágenes de contenedores de aprendizaje profundo se organiza por tipo de marco de trabajo.

Framework Procesador Nombres de imágenes de contenedores
Base GPU gcr.io/deeplearning-platform-release/base-cu110
gcr.io/deeplearning-platform-release/base-cu113
CPU gcr.io/deeplearning-platform-release/base-cpu
TensorFlow Enterprise 2.x GPU gcr.io/deeplearning-platform-release/tf-gpu
gcr.io/deeplearning-platform-release/tf-gpu.2-1
gcr.io/deeplearning-platform-release/tf-gpu.2-3
gcr.io/deeplearning-platform-release/tf-gpu.2-6
gcr.io/deeplearning-platform-release/tf-gpu.2-7
CPU gcr.io/deeplearning-platform-release/tf-cpu
gcr.io/deeplearning-platform-release/tf-cpu.2-1
gcr.io/deeplearning-platform-release/tf-cpu.2-3
gcr.io/deeplearning-platform-release/tf-cpu.2-6
gcr.io/deeplearning-platform-release/tf-cpu.2-7
TensorFlow Enterprise 1.x GPU gcr.io/deeplearning-platform-release/tf-gpu.1-15
CPU gcr.io/deeplearning-platform-release/tf-cpu.1-15
PyTorch GPU gcr.io/deeplearning-platform-release/pytorch-gpu
gcr.io/deeplearning-platform-release/pytorch-gpu.1-10
PyTorch/XLA TPU/GPU/CPU (experimental) gcr.io/deeplearning-platform-release/pytorch-xla
gcr.io/deeplearning-platform-release/pytorch-xla.1-10
R CPU (experimental) gcr.io/deeplearning-platform-release/r-cpu
gcr.io/deeplearning-platform-release/r-cpu.4-1
Scikit-learn CPU (experimental) gcr.io/deeplearning-platform-release/sklearn-cpu
gcr.io/deeplearning-platform-release/sklearn-cpu.0-23
XGBoost CPU (experimental) gcr.io/deeplearning-platform-release/xgboost-cpu
gcr.io/deeplearning-platform-release/xgboost-cpu.1-1
Imágenes de contenedor de aprendizaje profundo obsoletas Mixta gcr.io/deeplearning-platform-release/base-cu100
gcr.io/deeplearning-platform-release/base-cu101
gcr.io/deeplearning-platform-release/tf2-gpu.2-0
gcr.io/deeplearning-platform-release/tf2-cpu.2-0
gcr.io/deeplearning-platform-release/tf2-gpu.2-2
gcr.io/deeplearning-platform-release/tf2-cpu.2-2
gcr.io/deeplearning-platform-release/tf2-gpu.2-4
gcr.io/deeplearning-platform-release/tf2-cpu.2-4
gcr.io/deeplearning-platform-release/tf-gpu.2-5
gcr.io/deeplearning-platform-release/tf-cpu.2-5
gcr.io/deeplearning-platform-release/tf-gpu.1-13
gcr.io/deeplearning-platform-release/tf-gpu.1-14
gcr.io/deeplearning-platform-release/tf-cpu.1-13
gcr.io/deeplearning-platform-release/tf-cpu.1-14
gcr.io/deeplearning-platform-release/pytorch-gpu.1-0
gcr.io/deeplearning-platform-release/pytorch-gpu.1-1
gcr.io/deeplearning-platform-release/pytorch-gpu.1-2
gcr.io/deeplearning-platform-release/pytorch-gpu.1-3
gcr.io/deeplearning-platform-release/pytorch-gpu.1-4
gcr.io/deeplearning-platform-release/pytorch-gpu.1-6
gcr.io/deeplearning-platform-release/pytorch-gpu.1-7
gcr.io/deeplearning-platform-release/pytorch-gpu.1-8
gcr.io/deeplearning-platform-release/pytorch-gpu.1-9
gcr.io/deeplearning-platform-release/pytorch-cpu.1-0
gcr.io/deeplearning-platform-release/pytorch-cpu.1-1
gcr.io/deeplearning-platform-release/pytorch-cpu.1-2
gcr.io/deeplearning-platform-release/pytorch-cpu.1-3
gcr.io/deeplearning-platform-release/pytorch-cpu.1-4
gcr.io/deeplearning-platform-release/pytorch-xla.1-6
gcr.io/deeplearning-platform-release/pytorch-xla.1-7
gcr.io/deeplearning-platform-release/pytorch-xla.1-8
gcr.io/deeplearning-platform-release/pytorch-xla.1-9

Dependencias incluidas

Las listas de las dependencias de Python que se incluyen en cada versión están disponibles en Cloud Storage en

 gs://deeplearning-platform-release/installed-dependencies/containers/RELEASE_MILESTONE

Reemplaza RELEASE_MILESTONE por el evento importante de actualización, como m88. Por ejemplo, las listas para la actualización de M88 se encuentran en gs://deeplearning-platform-release/installed-dependencies/containers/m88/.

Imágenes de contenedor de TensorFlow Enterprise

Las imágenes de contenedor de TensorFlow Enterprise proporcionan una distribución optimizada de Google Cloud para TensorFlow, y las versiones específicas de esta distribución también incluyen asistencia a versiones de largo plazo. Para obtener más información sobre TensorFlow Enterprise, consulta la descripción general de TensorFlow Enterprise.

Imágenes experimentales

Algunas familias de imágenes de contenedores de aprendizaje profundo son experimentales, como lo indica la tabla de familias de imágenes. Las imágenes experimentales son compatibles en base al mejor esfuerzo y es posible que no reciban 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, puedes buscar la lista completa de imágenes de contenedor disponibles. Para crear una lista de todas las imágenes de contenedores de aprendizaje profundo disponibles, usa el siguiente comando en la herramienta de línea de comandos de gcloud con tu terminal preferida o en Cloud Shell.

gcloud container images list --repository="gcr.io/deeplearning-platform-release"

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?