지원되는 프레임워크 및 AWS 리전 - Amazon SageMaker

기계 번역으로 제공되는 번역입니다. 제공된 번역과 원본 영어의 내용이 상충하는 경우에는 영어 버전이 우선합니다.

지원되는 프레임워크 및 AWS 리전

SageMaker 스마트 체이프팅 데이터 로더를 사용하기 전에 선택한 프레임워크가 지원되는지, AWS 계정에서 인스턴스 유형을 사용할 수 있는지, AWS 계정이 지원되는 AWS 리전 중 하나에 있는지 확인합니다.

참고

SageMaker 스마트 체이프팅은 기존 데이터 병렬 처리 및 분산 데이터 병렬 처리를 통한 모델 PyTorch 훈련을 지원하므로 모든 GPU 작업자에서 모델 복제본을 만들고 AllReduce 작업을 사용합니다. 샤딩된 데이터 병렬 처리를 포함한 모델 병렬 처리 기법에서는 작동하지 않습니다. SageMaker 스마트 체이프팅은 데이터 병렬 처리 작업에서 작동하므로 훈련하는 모델이 각 GPU 메모리에 맞는지 확인하세요.

지원되는 프레임워크

SageMaker 스마트 시프팅은 다음과 같은 딥 러닝 프레임워크를 지원하며 AWS 딥 러닝 컨테이너를 통해 사용할 수 있습니다.

주제

PyTorch

프레임워크 프레임워크 버전 딥 러닝 컨테이너 URI
PyTorch 2.1.0

763104351884.dkr.ecr.region.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker

사전 구축된 컨테이너에 대한 자세한 내용은 Deep Learning SageMaker Containers 리포지토리의 프레임워크 컨테이너를 참조하세요. AWS GitHub

AWS 리전

SageMaker 스마트 체질 라이브러리와 함께 패키징된 컨테이너AWS Deep Learning Containers가 사용 AWS 리전 중인 에서 사용할 수 있습니다.

인스턴스 타입

모든 인스턴스 유형의 모든 PyTorch 훈련 작업에 SageMaker 스마트 체이프팅을 사용할 수 있습니다. P4d, P4de 또는 P5 인스턴스를 사용하는 것이 좋습니다.