Contenedores previamente compilados para el entrenamiento personalizado

Vertex AI proporciona imágenes de contenedor de Docker que puedes ejecutar como contenedores previamente compilados para el entrenamiento personalizado. Estos contenedores, que están organizados por framework de aprendizaje automático (AA) y versión de framework, incluyen dependencias comunes que tal vez quieras usar en el código de entrenamiento. A menudo, el uso de un contenedor previamente compilado es más simple que crear tu propio contenedor personalizado para el entrenamiento.

En este documento, se enumeran los contenedores previamente compilados para el entrenamiento y se describe cómo usarlos con una aplicación de entrenamiento de Python.

Vertex AI admite cada versión de framework según un programa para minimizar las vulnerabilidades de seguridad. Revisa la política de asistencia del framework de Vertex AI para comprender las implicaciones de las fechas de finalización de la asistencia y de finalización de la disponibilidad.

Imágenes de contenedor disponibles

Cada una de las siguientes imágenes de contenedor está disponible en varios repositorios de Artifact Registry, que almacenan datos en varias ubicaciones. Puedes usar cualquiera de los URI de una imagen cuando realices un entrenamiento personalizado; cada uno proporciona la misma imagen de contenedor. Cuando usas la consola de Google Cloud para realizar un entrenamiento personalizado, la consola de Google Cloud selecciona el URI que mejor coincide con la ubicación en la que usas Vertex AI. Esto reduce la latencia.

TensorFlow

Versión del marco de trabajo de AA	Aceleradores compatibles (y la versión de CUDA, si corresponde)	Fecha de finalización del parche y la compatibilidad	Fin de la disponibilidad	Imágenes compatibles
2.15 (Python 3.10)	Solo TPU	23 de octubre de 2024	23 de octubre de 2025	us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-15.cp310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.15.x (latest patch) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.14 (Python 3.10)	Solo CPU	26 de septiembre de 2024	26 de septiembre de 2025	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-14.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.14.x (latest patch) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.14 (Python 3.10)	GPU (CUDA 11.x)	26 de septiembre de 2024	26 de septiembre de 2025	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-14.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.14.x (latest patch) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.13 (Python 3.10)	Solo CPU	5 de julio de 2024	5 de julio de 2025	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-13.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.13.x (latest patch) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.13 (Python 3.10)	GPU (CUDA 11.x)	5 de julio de 2024	5 de julio de 2025	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-13.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.13.x (latest patch) sympy 1.12 xgboost 2.0.3 python-json-logger 2.0.7 WebOb 1.8.7 Paste 3.7.1 webapp2 3.0.0b1 mock 5.1.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.12 (Python 3.10)	Solo CPU	30 de junio de 2024	30 de junio de 2025	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-12.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.12.x (parche más reciente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.12 (Python 3.10)	GPU (CUDA 11.x)	30 de junio de 2024	30 de junio de 2025	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-12.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.12.x (parche más reciente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.12 (Python 3.8)	Solo TPU	30 de junio de 2024	30 de junio de 2025	us-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest europe-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest asia-docker.pkg.dev/vertex-ai/training/tf-tpu.2-12:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.12.x (parche más reciente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.11 (Python 3.10)	Solo CPU	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.11.x (parche más reciente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.11 (Python 3.10)	GPU (CUDA 11.x)	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.11.x (parche más reciente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.11 (Python 3.7)	Solo CPU	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-11:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.11.x (parche más reciente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.11 (Python 3.7)	GPU (CUDA 11.x)	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-11:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.11.x (parche más reciente) sympy 1.10.1 xgboost 1.6.2 python-json-logger 2.0.4 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 5.0.0 google-cloud-resource-manager 1.6.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.9 (Python 3.10)	Solo CPU	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.9.x (parche más reciente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.9 (Python 3.10)	GPU (CUDA 11.x)	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.9.x (parche más reciente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.9 (Python 3.7)	Solo CPU	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-9:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.9.x (parche más reciente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.9 (Python 3.7)	GPU (CUDA 11.x)	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-9:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.9.x (parche más reciente) sympy 1.10.1 xgboost 1.6.1 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.5.0 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.8 (Python 3.10)	Solo CPU	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.8.x (parche más reciente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.8 (Python 3.10)	GPU (CUDA 11.x)	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.8.x (parche más reciente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.8 (Python 3.7)	Solo CPU	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-8:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.8.x (parche más reciente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.8 (Python 3.7)	GPU (CUDA 11.x)	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-8:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.8.x (parche más reciente) sympy 1.9 xgboost 1.5.2 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.3 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2 Otras Dependencias de Deep Learning Containers
2.7 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-7:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.7.x (parche más reciente) sympy 1.9 xgboost 1.5.0 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.7 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-7:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.7.x (parche más reciente) sympy 1.9 xgboost 1.5.0 python-json-logger 2.0.2 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 1.3.1 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.6 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-6:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.6.x (parche más reciente) sympy 1.8 xgboost 1.4.2 python-json-logger 2.0.2 WebOb 1.8.7 Pega 3.5.0 webapp2 3.0.0b1 simulado 4.0.3 google-cloud-resource- Manager 1.0.2 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.6 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-6:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.6.x (parche más reciente) sympy 1.8 xgboost 1.4.2 python-json-logger 2.0.2 WebOb 1.8.7 Pega 3.5.0 webapp2 3.0.0b1 simulado 4.0.3 google-cloud-resource- Manager 1.0.2 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.5 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-5:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.5.x (parche más reciente) sympy 1.8 xgboost 1.4.0 PyYAML 3.13 python-json-logger 2.0.1 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 0.30.3 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.5 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-5:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.5.x (parche más reciente) sympy 1.8 xgboost 1.4.0 PyYAML 3.13 python-json-logger 2.0.1 WebOb 1.8.7 Paste 3.5.0 webapp2 3.0.0b1 mock 4.0.3 google-cloud-resource-manager 0.30.3 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.4 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-4:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.4.x (parche más reciente) numpy 1.19.4 pandas 1.1.5 scipy 1.5.4 scikit-learn 0.24.0 sympy 1.7.1 statsmodels 0.12.1 xgboost 1.3.1 oauth2client 4.1.3 httplib2 0.15.0 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 3.13 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.0 python-json-logger 2.0.1 wheel 0.36.2 WebOb 1.8.6 Paste 3.5.0 tornado 5.1.1 grpcio 1.32.0 requests 2.25.1 webapp2 3.0.0b1 mock 4.0.3 google-cloud-bigquery 2.6.1 google-cloud-bigtable 1.6.1 google-cloud-datastore 2.1.0 google-cloud-logging 1.15.0 google-cloud-pubsub 2.2.0 google-cloud-resource-manager 0.30.3 google-cloud-storage 1.35.0 joblib 1.0.0 cloudml-hypertune psutil 5.8.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.4 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-4:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.4.x (parche más reciente) numpy 1.19.4 pandas 1.1.5 scipy 1.5.4 scikit-learn 0.24.0 sympy 1.7.1 statsmodels 0.12.1 xgboost 1.3.1 oauth2client 4.1.3 httplib2 0.15.0 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 3.13 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.0 python-json-logger 2.0.1 wheel 0.36.2 WebOb 1.8.6 Paste 3.5.0 tornado 5.1.1 grpcio 1.32.0 requests 2.25.1 webapp2 3.0.0b1 mock 4.0.3 google-cloud-bigquery 2.6.1 google-cloud-bigtable 1.6.1 google-cloud-datastore 2.1.0 google-cloud-logging 1.15.0 google-cloud-pubsub 2.2.0 google-cloud-resource-manager 0.30.3 google-cloud-storage 1.35.0 joblib 1.0.0 cloudml-hypertune psutil 5.8.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.3 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-3:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.3.x (parche más reciente) numpy 1.18.5 pandas 1.1.3 scipy 1.4.1 scikit-learn 0.23.2 sympy 1.6.2 statsmodels 0.12.0 xgboost 1.2.1 oauth2client 4.1.3 httplib2 0.15.0 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 3.13 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.0 python-json-logger 2.0.1 wheel 0.35.1 WebOb 1.8.6 Paste 3.5.0 tornado 5.1.1 grpcio 1.33.1 requests 2.24.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 2.2.0 google-cloud-bigtable 1.5.1 google-cloud-datastore 1.15.3 google-cloud-logging 1.15.0 google-cloud-pubsub 2.1.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.32.0 joblib 0.17.0 cloudml-hypertune psutil 5.7.2 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.3 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-3:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.3.x (parche más reciente) numpy 1.18.5 pandas 1.1.3 scipy 1.4.1 scikit-learn 0.23.2 sympy 1.6.2 statsmodels 0.12.0 xgboost 1.2.1 oauth2client 4.1.3 httplib2 0.15.0 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 3.13 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.0 python-json-logger 2.0.1 wheel 0.35.1 WebOb 1.8.6 Paste 3.5.0 tornado 5.1.1 grpcio 1.33.1 requests 2.24.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 2.2.0 google-cloud-bigtable 1.5.1 google-cloud-datastore 1.15.3 google-cloud-logging 1.15.0 google-cloud-pubsub 2.1.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.32.0 joblib 0.17.0 cloudml-hypertune psutil 5.7.2 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.2 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.2.x (parche más reciente) numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 0.23.1 sympy 1.6 statsmodels 0.11.1 xgboost 1.1.1 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune psutil 5.7.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.2 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-2:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.2.x (parche más reciente) numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 0.23.1 sympy 1.6 statsmodels 0.11.1 xgboost 1.1.1 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune psutil 5.7.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.1 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-1:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.1.0 numpy 1.18.0 pandas 0.25.3 scipy 1.4.1 scikit-learn 0.22.1 sympy 1.5 statsmodels 0.10.2 xgboost 0.90 oauth2client 4.1.3 httplib2 0.15.0 python-dateutil 2.8.1 six 1.13.0 future 0.18.2 PyYAML 5.2 wrapt 1.11.2 crcmod 1.7 google-api-python-client 1.7.11 python-json-logger 0.1.11 wheel 0.33.6 WebOb 1.8.5 Paste 3.2.3 tornado 5.1.1 grpcio 1.26.0 requests 2.22.0 webapp2 3.0.0b1 mock 3.0.5 google-cloud-bigquery 1.23.1 google-cloud-bigtable 1.2.0 google-cloud-datastore 1.10.0 google-cloud-logging 1.14.0 google-cloud-pubsub 1.1.0 google-cloud-resource-manager 0.30.0 google-cloud-storage 1.23.0 joblib 0.14.1 cloudml-hypertune psutil 5.7.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.1 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-1:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 2.1.0 numpy 1.18.0 pandas 0.25.3 scipy 1.4.1 scikit-learn 0.22.1 sympy 1.5 statsmodels 0.10.2 xgboost 0.90 oauth2client 4.1.3 httplib2 0.15.0 python-dateutil 2.8.1 six 1.13.0 future 0.18.2 PyYAML 5.2 wrapt 1.11.2 crcmod 1.7 google-api-python-client 1.7.11 python-json-logger 0.1.11 wheel 0.33.6 WebOb 1.8.5 Paste 3.2.3 tornado 5.1.1 grpcio 1.26.0 requests 2.22.0 webapp2 3.0.0b1 mock 3.0.5 google-cloud-bigquery 1.23.1 google-cloud-bigtable 1.2.0 google-cloud-datastore 1.10.0 google-cloud-logging 1.14.0 google-cloud-pubsub 1.1.0 google-cloud-resource-manager 0.30.0 google-cloud-storage 1.23.0 joblib 0.14.1 cloudml-hypertune psutil 5.7.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.15 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest europe-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest asia-docker.pkg.dev/vertex-ai/training/tf-cpu.1-15:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 1.15.0 numpy 1.16.5 pandas 0.24.2 scipy 1.2.2 scikit-learn 0.20.4 sympy 1.4 statsmodels 0.10.1 xgboost 0.82 oauth2client 4.1.3 httplib2 0.12.0 python-dateutil 2.7.5 argparse 1.4.0 six 1.12.0 future 0.17.1 PyYAML 3.13 wrapt 1.11.1 crcmod 1.7 google-api-python-client 1.7.8 python-json-logger 0.1.10 wheel 0.32.3 WebOb 1.8.5 Paste 3.0.6 tornado 5.1.1 grpcio 1.18.0 requests 2.19.0 webapp2 3.0.0b1 mock 2.0.0 google-cloud-bigquery 1.20.0 google-cloud-bigtable 1.0.0 google-cloud-datastore 1.9.0 google-cloud-logging 1.12.1 google-cloud-pubsub 1.0.0 google-cloud-resource-manager 0.29.2 google-cloud-storage 1.19.1 joblib 0.13.0 cloudml-hypertune psutil 5.7.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.15 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest europe-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest asia-docker.pkg.dev/vertex-ai/training/tf-gpu.1-15:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu tensorflow 1.15.0 numpy 1.16.5 pandas 0.24.2 scipy 1.2.2 scikit-learn 0.20.4 sympy 1.4 statsmodels 0.10.1 xgboost 0.82 oauth2client 4.1.3 httplib2 0.12.0 python-dateutil 2.7.5 argparse 1.4.0 six 1.12.0 future 0.17.1 PyYAML 3.13 wrapt 1.11.1 crcmod 1.7 google-api-python-client 1.7.8 python-json-logger 0.1.10 wheel 0.32.3 WebOb 1.8.5 Paste 3.0.6 tornado 5.1.1 grpcio 1.18.0 requests 2.19.0 webapp2 3.0.0b1 mock 2.0.0 google-cloud-bigquery 1.20.0 google-cloud-bigtable 1.0.0 google-cloud-datastore 1.9.0 google-cloud-logging 1.12.1 google-cloud-pubsub 1.0.0 google-cloud-resource-manager 0.29.2 google-cloud-storage 1.19.1 joblib 0.13.0 cloudml-hypertune psutil 5.7.0 curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

PyTorch

Versión del marco de trabajo de AA	Aceleradores compatibles (y la versión de CUDA, si corresponde)	Fecha de finalización del parche y la compatibilidad	Fin de la disponibilidad	Imágenes compatibles
2.3 (Python 3.10)	Solo CPU	24 de abril, 2025	24 de abril, 2026	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-3.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.3.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.3 (Python 3.10)	Solo GPU (CUDA 12.x)	24 de abril, 2025	24 de abril, 2026	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-3.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.3.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.2 (Python 3.10)	Solo CPU	30 de enero de 2025	23 de octubre de 2026	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-2.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.2.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.2 (Python 3.10)	Solo GPU (CUDA 12.x)	30 de enero de 2025	30 de enero de 2026	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-2.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.2.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.1 (Python 3.10)	Solo CPU	4 de octubre de 2024	4 de octubre de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-1.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.1.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.1 (Python 3.10)	Solo GPU (CUDA 12.x)	4 de octubre de 2024	4 de octubre de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-1.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.1.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.1 (Python 3.8)	Solo TPU	23 de octubre de 2024	23 de octubre de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-1.cp310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.1.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.0 (Python 3.10)	Solo CPU	15 de marzo de 2024	15 de marzo de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-0.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.0.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.0 (Python 3.10)	Solo GPU (CUDA 11.x)	15 de marzo de 2024	15 de marzo de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-0.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.0.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
2.0 (Python 3.8)	Solo TPU	15 de mayo de 2024	15 de mayo de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-0:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 2.0.x (el parche más reciente) absl-py 2.1.0 cloud-tpu-client 0.10 python-json-logger 2.0.7 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.8 python3-dev python3.7-dev python3-pip python3-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.13 (Python 3.10)	CPU o GPU (CUDA 11.x)	15 de mayo de 2024	15 de mayo de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.13.x (parche más reciente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Otro Dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.13 (Python 3.7)	CPU o GPU (CUDA 11.x)	8 de diciembre de 2023	8 de diciembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-13:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.13.x (parche más reciente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Otro Dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.12 (Python 3.10)	CPU o GPU (CUDA 11.x)	15 de mayo de 2024	15 de mayo de 2025	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12.py310:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.12.x (parche más reciente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Otro Dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.12 (Python 3.7)	CPU o GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-12:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.12.x (parche más reciente) absl-py 1.3.0 cloud-tpu-client 0.10 python-json-logger 2.0.4 cloudml-hypertune Otro Dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.11 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-11:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.11.x (parche más reciente) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Otro Dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.11 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-11:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.11.x (parche más reciente) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Otro Dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.10 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-10:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.10.x (latest patch) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Other dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.10 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-10:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.10.x (latest patch) absl-py 1.0.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Other dependencias de contenedores de aprendizaje profundo ca-certificates-java libatlas-base-dev liblapack-dev gfortran gdb openjdk-8-jdk openjdk-8-jre-headless g++ libio-all-perl libyaml-0-2
1.9 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-9:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.9.x (parche más reciente) absl-py 0.13.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.9 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-9:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu PyTorch 1.9.x (parche más reciente) absl-py 0.13.0 cloud-tpu-client 0.10 python-json-logger 2.0.2 cloudml-hypertune Otras dependencias de contenedores de aprendizaje profundo curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.7 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-7:latest Dependencias incluidas Dependencias de contenedores de aprendizaje profundo
1.7 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-7:latest Dependencias incluidas Dependencias de contenedores de aprendizaje profundo
1.6 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.1-6:latest Dependencias incluidas Dependencias de contenedores de aprendizaje profundo
1.6 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-6:latest Dependencias incluidas Dependencias de contenedores de aprendizaje profundo
1.4 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-cpu.1-4:latest Dependencias incluidas Dependencias de contenedores de aprendizaje profundo
1.4 (Python 3.7)	GPU (CUDA 11.x)	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.1-4:latest Dependencias incluidas Dependencias de contenedores de aprendizaje profundo

scikit-learn

Versión del marco de trabajo de AA	Aceleradores compatibles (y la versión de CUDA, si corresponde)	Fecha de finalización del parche y la compatibilidad	Fin de la disponibilidad	Imágenes compatibles
1.0 (Python 3.10)	Solo CPU	30 de junio de 2024	30 Junio de 2025	us-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest europe-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest asia-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-0:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 1.0 sympy 1.6 statsmodels 0.11.1 xgboost 1.6.2 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.10 python3-dev python3.10-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.0 (Python 3.10)	GPU (CUDA 11.x)	30 de junio de 2024	30 Junio de 2025	us-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest europe-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest asia-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-0:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 1.0 sympy 1.6 statsmodels 0.11.1 xgboost 1.6.2 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.10 python3-dev python3.10-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
0.23 (Python 3.7)	Solo CPU	1 de septiembre de 2023	1 de septiembre de 2024	us-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest europe-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest asia-docker.pkg.dev/vertex-ai/training/scikit-learn-cpu.0-23:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 0.23.1 sympy 1.6 statsmodels 0.11.1 xgboost 1.1.1 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

XGBoost

Versión del marco de trabajo de AA	Aceleradores compatibles (y la versión de CUDA, si corresponde)	Fecha de finalización del parche y la compatibilidad	Fin de la disponibilidad	Imágenes compatibles
1.6 (Python 3.10)	Solo CPU	30 de junio de 2024	30 Junio de 2025	us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-6:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 0.23.1 sympy 1.6 statsmodels 0.11.1 xgboost 1.6.2 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.10 python3-dev python3.10-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.6 (Python 3.10)	GPU (CUDA 11.x)	30 de junio de 2024	30 Junio de 2025	us-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest europe-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest asia-docker.pkg.dev/vertex-ai/training/xgboost-gpu.1-6:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 0.23.1 sympy 1.6 statsmodels 0.11.1 xgboost 1.6.2 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.10 python3-dev python3.10-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv
1.1 (Python 3.7)	Solo CPU	15 de noviembre de 2023	15 de noviembre de 2024	us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.1-1:latest Dependencias incluidas Paquetes PyPI Paquetes de Ubuntu numpy 1.18.5 pandas 1.0.4 scipy 1.4.1 scikit-learn 0.23.1 sympy 1.6 statsmodels 0.11.1 xgboost 1.1.1 oauth2client 4.1.3 httplib2 0.18.1 python-dateutil 2.8.1 six 1.15.0 future 0.18.2 PyYAML 5.3.1 wrapt 1.12.1 crcmod 1.7 google-api-python-client 1.9.3 python-json-logger 0.1.11 wheel 0.34.2 WebOb 1.8.6 Paste 3.4.1 tornado 5.1.1 grpcio 1.29.0 requests 2.23.0 webapp2 3.0.0b1 mock 4.0.2 google-cloud-bigquery 1.25.0 google-cloud-bigtable 1.2.1 google-cloud-datastore 1.12.0 google-cloud-logging 1.15.0 google-cloud-pubsub 1.6.0 google-cloud-resource-manager 0.30.2 google-cloud-storage 1.29.0 joblib 0.15.1 cloudml-hypertune curl libcurl3-dev wget zip unzip git vim build-essential ca-certificates pkg-config rsync ca-certificates-java libatlas-base-dev liblapack-dev gfortran libpng-dev libjpeg-dev python3.7 python3-dev python3.7-dev python3-pip python3-setuptools python2.7 python-dev python-setuptools gdb openjdk-8-jdk openjdk-8-jre-headless g++ zlib1g-dev libio-all-perl module-init-tools libsnappy-dev libyaml-0-2 python-opencv

Entrena con contenedores previamente compilados

Si deseas ver un ejemplo de cómo usar un contenedor previamente compilado para el entrenamiento como parte de un flujo de trabajo más integral, ejecuta el notebook de Jupyter "Entrenamiento personalizado y predicción en línea” en uno de los siguientes entornos:

Abrir en Colab |Abrir en Colab Enterprise |Abrir en notebooks administrados por el usuario de Vertex AI Workbench |Ver en GitHub

Para usar un contenedor previamente compilado, consulta la guía de configuración de los contenedores.

Si usas una imagen de contenedor compatible con GPU, asegúrate de especificar los campos machineSpec.acceleratorType y machineSpec.acceleratorCount cuando crees tu recurso CustomJob, HyperparameterTuningJob o TrainingPipeline.

¿Qué sigue?

• Obtén más información sobre cómo crear una aplicación de entrenamiento de Python a fin de usar con un contenedor previamente compilado.