개요

Sagemaker Studio에 전용 도커 이미지를 등록하는 방법을 정리해둔다.

전제

ECR에 이미 Docker 이미지는 Push되어 있다고 가정한다.

Custom Image 등록방법

Images 란을 클릭한다.

Create Image를 선택한다.

ECR 경로를 입력한다.

Image Sagemaker Studio에 Attach하기

이후 끝이 아니라 등록한 커스텀 이미지를 Attach해주어야 한다. Control Panel을 클릭한다.

Attach Image를 클릭한다.

기존 이미지로 내가 원하는 이미지와 버전을 선택한다.

kernel 이름은 편의상 python3, python3로 통일한다.

검증하기

Sagemaker Studio에 아래와 같이 커스텀 이미지가 정상 등록된 것을 확인 할 수 있다.