사전 빌드된 커스텀 학습용 컨테이너

Vertex AI는 커스텀 학습을 위해 사전 빌드된 컨테이너로서 실행되는 Docker 컨테이너 이미지를 제공합니다. 머신러닝(ML) 프레임워크 및 프레임워크 버전으로 구성된 이러한 컨테이너에는 학습 코드에서 사용하려는 일반적인 종속 항목이 포함됩니다. 종종 사전 빌드된 컨테이너를 사용하는 것이 학습용 커스텀 컨테이너를 만드는 것보다 간단합니다.

이 문서에서는 학습용으로 사전 빌드된 컨테이너를 나열하고 Python 학습 애플리케이션과 함께 사용하는 방법을 설명합니다.

Vertex AI는 보안 취약점을 최소화하도록 일정에 따라 각 프레임워크 버전을 지원합니다. Vertex AI 프레임워크 지원 정책을 검토하여 지원 종료와 가용성 날짜 종료가 미치는 영향을 파악합니다.

사용 가능한 컨테이너 이미지

다음 각 컨테이너 이미지는 여러 Artifact Registry 저장소에서 제공되며, 이러한 저장소는 데이터를 다양한 위치에 저장합니다. 커스텀 학습을 수행할 때 이미지의 URI 중 어느 것이나 사용할 수 있습니다. 모두 동일한 컨테이너 이미지를 제공합니다. Google Cloud 콘솔을 사용하여 커스텀 학습을 수행하는 경우, Google Cloud 콘솔은 Vertex AI를 사용 중인 위치에 가장 적합한 URI를 선택합니다. 이렇게 하면 지연 시간이 줄어듭니다.

TensorFlow

ML 프레임워크 버전 지원되는 가속기(해당하는 경우 CUDA 버전) 패치 및 지원 종료일 지원 종료 지원되는 이미지
2.15(Python 3.10) TPU만 해당 2024년 10월 23일 2025년 10월 23일
  • 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만 2024년 9월 26일 2025년 9월 26일
  • 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) 2024년 9월 26일 2025년 9월 26일
  • 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만 2024년 7월 5일 2025년 7월 5일
  • 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) 2024년 7월 5일 2025년 7월 5일
  • 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만 2024년 6월 30일 2025년 6월 30일
  • 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) 2024년 6월 30일 2025년 6월 30일
  • 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만 해당 2024년 6월 30일 2025년 6월 30일
  • 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만 2023년 11월 15일 2024년 11월 15일
  • 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) 2023년 11월 15일 2024년 11월 15일
  • 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만 2023년 11월 15일 2024년 11월 15일
  • 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) 2023년 11월 15일 2024년 11월 15일
  • 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만 2023년 11월 15일 2024년 11월 15일
  • 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) 2023년 11월 15일 2024년 11월 15일
  • 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만 2023년 11월 15일 2024년 11월 15일
  • 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) 2023년 11월 15일 2024년 11월 15일
  • 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만 2023년 11월 15일 2024년 11월 15일
  • 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) 2023년 11월 15일 2024년 11월 15일
  • 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만 2023년 11월 15일 2024년 11월 15일
  • 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) 2023년 11월 15일 2024년 11월 15일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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 프레임워크 버전 지원되는 가속기(해당하는 경우 CUDA 버전) 패치 및 지원 종료일 지원 종료 지원되는 이미지
2.3(Python 3.10) CPU만 2025년 4월 24일 2026년 4월 24일
  • 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만(CUDA 12.x) 2025년 4월 24일 2026년 4월 24일
  • 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만 2025년 1월 30일 2026년 10월 23일
  • 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만(CUDA 12.x) 2025년 1월 30일 2026년 1월 30일
  • 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.1(Python 3.10) CPU만 2024년 10월 4일 2025년 10월 4일
  • 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만(CUDA 12.x) 2024년 10월 4일 2025년 10월 4일
  • 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만 해당 2024년 10월 23일 2025년 10월 23일
  • 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만 2024년 3월 15일 2025년 3월 15일
  • 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만(CUDA 11.x) 2024년 3월 15일 2025년 3월 15일
  • 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만 해당 2024년 5월 15일 2025년 5월 15일
  • 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 또는 GPU(CUDA 11.x) 2024년 5월 15일 2025년 5월 15일
  • 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 또는 GPU(CUDA 11.x) 2023년 12월 8일 2024년 12월 8일
  • 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 또는 GPU(CUDA 11.x) 2024년 5월 15일 2025년 5월 15일
  • 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 또는 GPU(CUDA 11.x) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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) 2023년 9월 1일 2024년 9월 1일
  • 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 프레임워크 버전 지원되는 가속기(해당하는 경우 CUDA 버전) 패치 및 지원 종료일 지원 종료 지원되는 이미지
1.0(Python 3.10) CPU만 2024년 6월 30일 2025년 6월 30일
  • 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) 2024년 6월 30일 2025년 6월 30일
  • 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만 2023년 9월 1일 2024년 9월 1일
  • 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 프레임워크 버전 지원되는 가속기(해당하는 경우 CUDA 버전) 패치 및 지원 종료일 지원 종료 지원되는 이미지
1.6(Python 3.10) CPU만 2024년 6월 30일 2025년 6월 30일
  • 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) 2024년 6월 30일 2025년 6월 30일
  • 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만 2023년 11월 15일 2024년 11월 15일
  • 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

사전 빌드된 컨테이너를 사용한 학습

사전 빌드된 컨테이너를 사용하려면 컨테이너 설정 구성 가이드를 참조하세요.

GPU를 지원하는 컨테이너 이미지를 사용하는 경우 CustomJob, HyperparameterTuningJob, TrainingPipeline 리소스를 만들 때 machineSpec.acceleratorTypemachineSpec.acceleratorCount 필드를 지정해야 합니다.

다음 단계