Console
    Contact Us Start free

    Prebuilt containers for custom training

    Vertex AI provides Docker container images that you run as prebuilt containers for custom training. These containers, which are organized by machine learning (ML) framework and framework version, include common dependencies that you might want to use in training code. Often, using a prebuilt container is simpler than creating your own custom container for training .

    This document lists the prebuilt containers for training, and it describes how to use them with a Python training application .

    Vertex AI supports each framework version based on a schedule to minimize security vulnerabilities. Review the Vertex AI framework support policy to understand the implications of the end-of-support and end-of-availability dates.

    Available container images

    Each of the following container images is available in several Artifact Registry repositories , which store data in various locations . You can use any of the URIs for an image when you perform custom training; each provides the same container image. When using Google Cloud console to perform custom training, the Google Cloud console selects the URI that best matches the location where you're using Vertex AI . This reduces latency.

    TensorFlow

    ML framework version
    Supported accelerators (and CUDA version, if applicable)
    End of patch and support date
    End of availability
    Supported images
    2.17 (Python 3.10)
    CPU only
    Jul 11, 2025
    Jul 11, 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)
    Jul 11, 2025
    Jul 11, 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)
    CPU only
    Jun 28, 2025
    Jun 28, 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)
    Jun 28, 2025
    Jun 28, 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)
    CPU only
    Nov 14, 2024
    Nov 14, 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)
    Nov 14, 2024
    Nov 14, 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)
    TPU only
    Oct 23, 2024
    Oct 23, 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)
    CPU only
    Sep 26, 2024
    Sep 26, 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
    Sep 26, 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)
    CPU only
    Jul 5, 2024
    Jul 5, 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)
    Jul 5, 2024
    Jul 5, 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)
    CPU only
    Jun 30, 2024
    Jun 30, 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)
    Jun 30, 2024
    Jun 30, 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)
    TPU only
    Jun 30, 2024
    Jun 30, 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)
    CPU only
    Nov 15, 2023
    Nov 15, 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)
    Nov 15, 2023
    Nov 15, 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)
    CPU only
    Nov 15, 2023
    Nov 15, 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)
    Nov 15, 2023
    Nov 15, 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)
    CPU only
    Nov 15, 2023
    Nov 15, 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)
    Nov 15, 2023
    Nov 15, 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)
    CPU only
    Nov 15, 2023
    Nov 15, 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)
    Nov 15, 2023
    Nov 15, 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)
    CPU only
    Nov 15, 2023
    Nov 15, 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)
    Nov 15, 2023
    Nov 15, 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)
    CPU only
    Nov 15, 2023
    Nov 15, 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)
    Nov 15, 2023
    Nov 15, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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

    ML framework version
    Supported accelerators (and CUDA version, if applicable)
    End of patch and support date
    End of availability
    Supported images
    2.4 (Python 3.10)
    CPU only
    Jul 24, 2025
    Jul 24, 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)
    GPU only (CUDA 12.x)
    Jul 24, 2025
    Jul 24, 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)
    CPU only
    Apr 24, 2025
    Apr 24, 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)
    GPU only (CUDA 12.x)
    Apr 24, 2025
    Apr 24, 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)
    CPU only
    Jan 30, 2025
    Oct 23, 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)
    GPU only (CUDA 12.x)
    Jan 30, 2025
    Jan 30, 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)
    TPU only
    Nov 6, 2024
    Nov 6, 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)
    CPU only
    Oct 4, 2024
    Oct 4, 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)
    GPU only (CUDA 12.x)
    Oct 4, 2024
    Oct 4, 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)
    TPU only
    Oct 23, 2024
    Oct 23, 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)
    CPU only
    Mar 15, 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)
    GPU only (CUDA 11.x)
    Mar 15, 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)
    TPU only
    May 15, 2024
    May 15, 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 or GPU (CUDA 11.x)
    May 15, 2024
    May 15, 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 or GPU (CUDA 11.x)
    Dec 8, 2023
    Dec 8, 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 or GPU (CUDA 11.x)
    May 15, 2024
    May 15, 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 or GPU (CUDA 11.x)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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)
    Sep 1, 2023
    Sep 1, 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

    ML framework version
    Supported accelerators (and CUDA version, if applicable)
    End of patch and support date
    End of availability
    Supported images
    1.6 (Python 3.10)
    CPU only
    Jul 11, 2025
    Jul 11, 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)
    Jul 11, 2025
    Jul 11, 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)
    CPU only
    June 30, 2024
    June 30, 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)
    June 30, 2024
    June 30, 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)
    CPU only
    Sep 1, 2023
    Sep 1, 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

    ML framework version
    Supported accelerators (and CUDA version, if applicable)
    End of patch and support date
    End of availability
    Supported images
    2.1 (Python 3.10)
    CPU only
    Jul 11, 2025
    Jul 11, 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)
    Jul 11, 2025
    Jul 11, 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)
    CPU only
    June 30, 2024
    June 30, 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)
    June 30, 2024
    June 30, 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)
    CPU only
    Nov 15, 2023
    Nov 15, 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

    Supported versions of Ray on Vertex AI

    Ray on Vertex AI container images are available. The following versions of prebuilt containers are supported:

    Ray version
    Supported accelerators (and CUDA version, if applicable)
    End of patch and support date
    End of availability
    Supported images
    2.47.1 (Python 3.11)
    CPU only
    Jul 28, 2026
    Jul 28, 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)
    Jul 28, 2026
    Jul 28, 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)
    CPU only
    Mar 1, 2026
    Mar 1, 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)
    Mar 1, 2026
    Mar 1, 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)
    CPU only
    Mar 1, 2026
    Mar 1, 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)
    Mar 1, 2026
    Mar 1, 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)
    CPU only
    September 16, 2025
    September 16, 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)
    September 16, 2025
    September 16, 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)
    CPU only
    May 7, 2025
    May 7, 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)
    May 7, 2025
    May 7, 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

    Train with a prebuilt container

    To use a prebuilt container, read the guide to configuring container settings .

    If you're using a container image that supports GPUs, make sure to specify the machineSpec.acceleratorType and machineSpec.acceleratorCount fields when you create your CustomJob , HyperparameterTuningJob , or TrainingPipeline resource.

    What's next
    Design a Mobile Site
    View Site in Mobile | Classic
    Share by: