Contenedores prediseñados para el entrenamiento personalizado

Vertex AI proporciona imágenes de contenedor Docker que puedes ejecutar como contenedores prediseñados para el entrenamiento personalizado. Estos contenedores, que se organizan por framework de aprendizaje automático (ML) y versión del framework, incluyen dependencias comunes que puedes usar en el código de entrenamiento. A menudo, usar un contenedor prediseñado es más sencillo que crear tu propio contenedor personalizado para el entrenamiento.

En este documento se enumeran los contenedores prediseñados para el entrenamiento y se describe cómo usarlos con una aplicación de entrenamiento de Python.

Vertex AI admite cada versión del framework según una programación para minimizar las vulnerabilidades de seguridad. Consulta la política de asistencia del framework de Vertex AI para conocer las implicaciones de las fechas de finalización de la asistencia y 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 URIs de una imagen cuando realices un entrenamiento personalizado. Todos proporcionan la misma imagen de contenedor. Cuando se usa la Google Cloud consola para realizar un entrenamiento personalizado, la Google Cloud consola selecciona el URI que mejor se adapte a la ubicación en la que estás usando Vertex AI. Esto reduce la latencia.

TensorFlow

Versión de framework de aprendizaje automático Aceleradores compatibles (y versión de CUDA, si procede) Fecha de fin de los parches y de la asistencia Fin de la disponibilidad Imágenes compatibles
2.17 (Python 3.10) Solo CPU 11 de julio del 2025 11 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-17.py310:latest
2.17 (Python 3.10) GPU (CUDA 12.x) 11 de julio del 2025 11 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-17.py310:latest
2.16 (Python 3.10) Solo CPU 28 de junio del 2025 28 de junio del 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-16.py310:latest
2.16 (Python 3.10) GPU (CUDA 12.x) 28 de junio del 2025 28 de junio del 2026
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-16.py310:latest
2.15 (Python 3.10) Solo CPU 14 de noviembre del 2024 14 nov. 2025
  • us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-cpu.2-15.py310:latest
2.15 (Python 3.10) GPU (CUDA 12.x) 14 de noviembre del 2024 14 nov. 2025
  • us-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/tf-gpu.2-15.py310:latest
2.15 (Python 3.10) Solo TPU 23 de octubre del 2024 23 de octubre del 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
2.14 (Python 3.10) Solo CPU Sep 26, 2024 26 de septiembre del 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
2.14 (Python 3.10) GPU (CUDA 11.x) Sep 26, 2024 26 de septiembre del 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
2.13 (Python 3.10) Solo CPU 5 de julio del 2024 5 de julio del 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
2.13 (Python 3.10) GPU (CUDA 11.x) 5 de julio del 2024 5 de julio del 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
2.12 (Python 3.10) Solo CPU 30 de junio del 2024 30 de junio del 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
2.12 (Python 3.10) GPU (CUDA 11.x) 30 de junio del 2024 30 de junio del 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
2.12 (Python 3.8) Solo TPU 30 de junio del 2024 30 de junio del 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
2.11 (Python 3.10) Solo CPU 15 de noviembre del 2023 15 de noviembre del 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
2.11 (Python 3.10) GPU (CUDA 11.x) 15 de noviembre del 2023 15 de noviembre del 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
2.11 (Python 3.7) Solo CPU 15 de noviembre del 2023 15 de noviembre del 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
2.11 (Python 3.7) GPU (CUDA 11.x) 15 de noviembre del 2023 15 de noviembre del 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
2.9 (Python 3.10) Solo CPU 15 de noviembre del 2023 15 de noviembre del 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
2.9 (Python 3.10) GPU (CUDA 11.x) 15 de noviembre del 2023 15 de noviembre del 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
2.9 (Python 3.7) Solo CPU 15 de noviembre del 2023 15 de noviembre del 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
2.9 (Python 3.7) GPU (CUDA 11.x) 15 de noviembre del 2023 15 de noviembre del 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
2.8 (Python 3.10) Solo CPU 15 de noviembre del 2023 15 de noviembre del 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
2.8 (Python 3.10) GPU (CUDA 11.x) 15 de noviembre del 2023 15 de noviembre del 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
2.8 (Python 3.7) Solo CPU 15 de noviembre del 2023 15 de noviembre del 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
2.8 (Python 3.7) GPU (CUDA 11.x) 15 de noviembre del 2023 15 de noviembre del 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
2.7 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
2.7 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
2.6 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
2.6 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
2.5 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
2.5 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
2.4 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
2.4 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
2.3 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
2.3 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
2.2 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
2.2 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
2.1 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
2.1 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
1.15 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
1.15 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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

PyTorch

Versión de framework de aprendizaje automático Aceleradores compatibles (y versión de CUDA, si procede) Fecha de fin de los parches y de la asistencia Fin de la disponibilidad Imágenes compatibles
2.4 (Python 3.10) Solo CPU 24 de julio del 2025 24 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-xla.2-4.py310:latest
2.4 (Python 3.10) Solo GPU (CUDA 12.x) 24 de julio del 2025 24 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-gpu.2-4.py310:latest
2.3 (Python 3.10) Solo CPU 24 de abril del 2025 24 de abril del 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
2.3 (Python 3.10) Solo GPU (CUDA 12.x) 24 de abril del 2025 24 de abril del 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
2.2 (Python 3.10) Solo CPU 30 de enero del 2025 23 de octubre del 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
2.2 (Python 3.10) Solo GPU (CUDA 12.x) 30 de enero del 2025 30 de enero del 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
2.2 (Python 3.10) Solo TPU 6 de noviembre del 2024 6 de noviembre del 2025
  • us-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
  • europe-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
  • asia-docker.pkg.dev/vertex-ai/training/pytorch-tpu.2-2.cp310:latest
2.1 (Python 3.10) Solo CPU 4 de octubre del 2024 4 de octubre del 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
2.1 (Python 3.10) Solo GPU (CUDA 12.x) 4 de octubre del 2024 4 de octubre del 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
2.1 (Python 3.8) Solo TPU 23 de octubre del 2024 23 de octubre del 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
2.0 (Python 3.10) Solo CPU 15 mar. 2024 Mar 15, 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
2.0 (Python 3.10) Solo GPU (CUDA 11.x) 15 mar. 2024 Mar 15, 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
2.0 (Python 3.8) Solo TPU 15 de mayo del 2024 15 de mayo del 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
1.13 (Python 3.10) CPU o GPU (CUDA 11.x) 15 de mayo del 2024 15 de mayo del 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
1.13 (Python 3.7) CPU o GPU (CUDA 11.x) 8 de diciembre del 2023 8 dic., 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
1.12 (Python 3.10) CPU o GPU (CUDA 11.x) 15 de mayo del 2024 15 de mayo del 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
1.12 (Python 3.7) CPU o GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
1.11 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
1.11 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
1.10 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
1.10 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
1.9 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
1.9 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
1.7 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
1.7 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
1.6 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
1.6 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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
1.4 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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
1.4 (Python 3.7) GPU (CUDA 11.x) 1 sep. 2023 1 de septiembre del 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

scikit‑learn

Versión de framework de aprendizaje automático Aceleradores compatibles (y versión de CUDA, si procede) Fecha de fin de los parches y de la asistencia Fin de la disponibilidad Imágenes compatibles
1.6 (Python 3.10) Solo CPU 11 de julio del 2025 11 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-cpu.1-6:latest
1.6 (Python 3.10) GPU (CUDA 12.x) 11 de julio del 2025 11 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
  • europe-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
  • asia-docker.pkg.dev/vertex-ai/training/sklearn-gpu.1-6:latest
1.0 (Python 3.10) Solo CPU 30 de junio del 2024 30 de junio del 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
1.0 (Python 3.10) GPU (CUDA 11.x) 30 de junio del 2024 30 de junio del 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
0.23 (Python 3.7) Solo CPU 1 sep. 2023 1 de septiembre del 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

XGBoost

Versión de framework de aprendizaje automático Aceleradores compatibles (y versión de CUDA, si procede) Fecha de fin de los parches y de la asistencia Fin de la disponibilidad Imágenes compatibles
2.1 (Python 3.10) Solo CPU 11 de julio del 2025 11 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-cpu.2-1:latest
2.1 (Python 3.10) GPU (CUDA 12.x) 11 de julio del 2025 11 de julio del 2026
  • us-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
  • europe-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
  • asia-docker.pkg.dev/vertex-ai/training/xgboost-gpu.2-1:latest
1.6 (Python 3.10) Solo CPU 30 de junio del 2024 30 de junio del 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
1.6 (Python 3.10) GPU (CUDA 11.x) 30 de junio del 2024 30 de junio del 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
1.1 (Python 3.7) Solo CPU 15 de noviembre del 2023 15 de noviembre del 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

Versiones compatibles de Ray en Vertex AI

Las imágenes de contenedor de Ray en Vertex AI están disponibles. Se admiten las siguientes versiones de contenedores precompilados:

Versión de Ray Aceleradores compatibles (y versión de CUDA, si procede) Fecha de fin de los parches y de la asistencia Fin de la disponibilidad Imágenes compatibles
2.47.1 (Python 3.11) Solo CPU 28 de julio del 2026 28 de julio del 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-47.py311:latest
2.47.1 (Python 3.11) GPU (CUDA 11.x) 28 de julio del 2026 28 de julio del 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-47.py311:latest
2.42.0 (Python 3.11) Solo CPU 1 de marzo del 2026 1 de marzo del 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py311:latest
2.42.0 (Python 3.11) GPU (CUDA 11.x) 1 de marzo del 2026 1 de marzo del 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py311:latest
2.42.0 (Python 3.10) Solo CPU 1 de marzo del 2026 1 de marzo del 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-42.py310:latest
2.42.0 (Python 3.10) GPU (CUDA 11.x) 1 de marzo del 2026 1 de marzo del 2027
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-42.py310:latest
2.33.0 (Python 3.10) Solo CPU del 16 de septiembre del 2025 16 de septiembre del 2026
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-33.py310:latest
2.33.0 (Python 3.10) GPU (CUDA 11.x) del 16 de septiembre del 2025 16 de septiembre del 2026
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-33.py310:latest
2.9.3 (Python 3.10) Solo CPU 7 de mayo del 2025 7 de mayo del 2026
  • us-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-cpu.2-9.py310:latest
2.9.3 (Python 3.10) GPU (CUDA 11.x) 7 de mayo del 2025 7 de mayo del 2026
  • us-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest
  • europe-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest
  • asia-docker.pkg.dev/vertex-ai/training/ray-gpu.2-9.py310:latest

Entrenar con un contenedor prediseñado

Para usar un contenedor prediseñado, consulta la guía para configurar los ajustes del contenedor.

Si usas una imagen de contenedor que admite GPUs, asegúrate de especificar los campos machineSpec.acceleratorType y machineSpec.acceleratorCount al crear tu recurso CustomJob, HyperparameterTuningJob o TrainingPipeline.

Siguientes pasos