컨테이너 이미지 선택

이 페이지는 사용할 컨테이너 이미지를 선택하는 데 도움이 됩니다.

컨테이너 이미지 유형 선택

각 컨테이너 이미지는 Python 3 환경을 제공하며 선택된 데이터 과학 프레임워크(예: PyTorch 또는 TensorFlow), Conda, GPU 이미지용 NVIDIA 스택(CUDA, cuDNN, NCCL2), 기타 여러 지원 패키지 및 도구를 포함합니다. 원하는 컨테이너 이미지를 찾으려면 아래 표를 참조하세요.

다음 Deep Learning Containers 이미지 유형 목록은 프레임워크 유형별로 정리되어 있습니다.

프레임워크 프로세서 컨테이너 이미지 이름
기본 GPU gcr.io/deeplearning-platform-release/base-cu100
gcr.io/deeplearning-platform-release/base-cu101
gcr.io/deeplearning-platform-release/base-cu110
CPU gcr.io/deeplearning-platform-release/base-cpu
TensorFlow Enterprise 2.x GPU gcr.io/deeplearning-platform-release/tf2-gpu.2-1
gcr.io/deeplearning-platform-release/tf2-gpu.2-3
gcr.io/deeplearning-platform-release/tf2-gpu.2-5
CPU gcr.io/deeplearning-platform-release/tf2-cpu.2-1
gcr.io/deeplearning-platform-release/tf2-cpu.2-3
gcr.io/deeplearning-platform-release/tf2-cpu.2-5
TensorFlow Enterprise 1.x GPU gcr.io/deeplearning-platform-release/tf-gpu
gcr.io/deeplearning-platform-release/tf-gpu.1-15
CPU gcr.io/deeplearning-platform-release/tf-cpu
gcr.io/deeplearning-platform-release/tf-cpu.1-15
TensorFlow 2.x GPU gcr.io/deeplearning-platform-release/tf2-gpu
gcr.io/deeplearning-platform-release/tf2-gpu.2-0
gcr.io/deeplearning-platform-release/tf2-gpu.2-2
gcr.io/deeplearning-platform-release/tf2-gpu.2-4
CPU gcr.io/deeplearning-platform-release/tf2-cpu
gcr.io/deeplearning-platform-release/tf2-cpu.2-0
gcr.io/deeplearning-platform-release/tf2-cpu.2-2
gcr.io/deeplearning-platform-release/tf2-cpu.2-4
TensorFlow 1.x GPU gcr.io/deeplearning-platform-release/tf-gpu.1-13
gcr.io/deeplearning-platform-release/tf-gpu.1-14
CPU gcr.io/deeplearning-platform-release/tf-cpu.1-13
gcr.io/deeplearning-platform-release/tf-cpu.1-14
PyTorch GPU gcr.io/deeplearning-platform-release/pytorch-gpu
gcr.io/deeplearning-platform-release/pytorch-gpu.1-0
gcr.io/deeplearning-platform-release/pytorch-gpu.1-1
gcr.io/deeplearning-platform-release/pytorch-gpu.1-2
gcr.io/deeplearning-platform-release/pytorch-gpu.1-3
gcr.io/deeplearning-platform-release/pytorch-gpu.1-4
gcr.io/deeplearning-platform-release/pytorch-gpu.1-6
gcr.io/deeplearning-platform-release/pytorch-gpu.1-7
gcr.io/deeplearning-platform-release/pytorch-gpu.1-8
CPU gcr.io/deeplearning-platform-release/pytorch-cpu
gcr.io/deeplearning-platform-release/pytorch-cpu.1-0
gcr.io/deeplearning-platform-release/pytorch-cpu.1-1
gcr.io/deeplearning-platform-release/pytorch-cpu.1-2
gcr.io/deeplearning-platform-release/pytorch-cpu.1-3
gcr.io/deeplearning-platform-release/pytorch-cpu.1-4
PyTorch XLA TPU/GPU/CPU (실험용) gcr.io/deeplearning-platform-release/pytorch-xla.1-6
gcr.io/deeplearning-platform-release/pytorch-xla.1-7
gcr.io/deeplearning-platform-release/pytorch-xla.1-8
R CPU (실험용) gcr.io/deeplearning-platform-release/r-cpu
gcr.io/deeplearning-platform-release/r-cpu.4-0
scikit-learn CPU (실험용) gcr.io/deeplearning-platform-release/sklearn-cpu
gcr.io/deeplearning-platform-release/sklearn-cpu.0-23
XGBoost CPU (실험용) gcr.io/deeplearning-platform-release/xgboost-cpu
gcr.io/deeplearning-platform-release/xgboost-cpu.1-1

TensorFlow Enterprise 컨테이너 이미지

TensorFlow Enterprise 컨테이너 이미지는 Google Cloud에 최적화된 TensorFlow 배포판을 제공하며 특정 버전의 TensorFlow Enterprise 배포판에는 장기 버전 지원도 포함되어 있습니다. TensorFlow Enterprise에 대해 자세히 알아보려면 TensorFlow Enterprise 개요를 읽어보세요.

실험용 이미지

이미지 계열 표에 명시된 대로 일부 Deep Learning Containers 이미지 계열은 실험용으로 제공됩니다. 실험용 이미지는 가능한 범위 내에서 최대한 지원되지만 프레임워크가 새로 출시되어도 업데이트되지 않을 수 있습니다.

사용 가능한 모든 버전 목록 표시

특정 프레임워크 또는 CUDA 버전이 필요한 경우 사용 가능한 컨테이너 이미지의 전체 목록을 검색할 수 있습니다. 사용 가능한 모든 Deep Learning Containers 이미지를 나열하려면 선호하는 터미널이 있는 gcloud 명령줄 도구 또는 Cloud Shell에서 다음 명령어를 사용합니다.

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

로컬에서 사용

Deep Learning Containers를 로컬에서 가져와 사용할 수 있습니다. 이렇게 하려면 로컬 Deep Learning Containers 시작하기를 참조하세요.

다음 단계

  • Deep Learning Containers 개요를 읽고 컨테이너 이미지에 사전 설치된 항목에 대해 자세히 알아봅니다.
  • Deep Learning Containers 이미지를 빌드하고 푸시하는 방법에 대한 안내를 제공하는 안내 가이드를 통해 Deep Learning Containers를 시작합니다.