(보관됨) SageMaker 모델 병렬 처리 라이브러리 v1.x - Amazon SageMaker AI

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

(보관됨) SageMaker 모델 병렬 처리 라이브러리 v1.x

중요

2023년 12월 19일부터 SageMaker 모델 병렬 처리(SMP) 라이브러리 v2가 릴리스됩니다. SMP 라이브러리 v2에 유리하게, SMP v1 기능은 향후 릴리스에서 더 이상 지원되지 않습니다. 다음 섹션 및 주제는 보관되어 있으며 SMP 라이브러리 v1 사용에 따라 다릅니다. SMP 라이브러리 v2 사용에 대한 자세한 내용은 SageMaker 모델 병렬 처리 라이브러리 v2 섹션을 참조하세요.

Amazon SageMaker AI의 모델 병렬 라이브러리를 사용하여 GPU 메모리 제한으로 인해 훈련하기 어려운 대규모 딥 러닝(DL) 모델을 훈련할 수 있습니다. 이 라이브러리는 한 모델을 여러 GPU 및 인스턴스로 자동으로, 그리고 효율적으로 분할합니다. 이 라이브러리를 사용하면 수십억 개 또는 수조 개의 파라미터를 포함하는 대규모 DL 모델을 효율적으로 훈련시켜 목표 예측 정확도를 더 빨리 달성할 수 있습니다.

이 라이브러리를 사용하면 코드 변경을 최소화하면서 사용자 고유의 TensorFlow 모델 및 PyTorch 모델을 여러 GPU와 여러 노드로 자동 분할할 수 있습니다. SageMaker Python SDK로 이 라이브러리의 API에 액세스할 수 있습니다.

다음 섹션들을 사용하여 모델 병렬 처리 및 SageMaker 모델 병렬 라이브러리에 대해 자세히 알아보세요. 이 라이브러리의 API 설명서는 SageMaker Python SDK v2.199.0 설명서분산 훈련 API에 수록되어 있습니다.