(아카이브) SageMaker 모델 병렬화 라이브러리 v1.x - 아마존 SageMaker

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

(아카이브) SageMaker 모델 병렬화 라이브러리 v1.x

중요

2023년 12월 19일부터 SageMaker 모델 병렬화 (SMP) 라이브러리 v2가 출시되었습니다. SMP 라이브러리 v2를 선호하여 SMP v1 기능은 향후 릴리스에서 더 이상 지원되지 않습니다. 다음 섹션 및 항목은 보관되어 있으며 SMP 라이브러리 v1 사용에 특화되어 있습니다. SMP 라이브러리 v2 사용에 대한 자세한 내용은 을 참조하십시오. SageMaker 모델 병렬화 라이브러리 v2

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

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

모델 병렬 처리 및 모델 병렬 라이브러리에 대해 자세히 알아보려면 다음 섹션을 참조하십시오. SageMaker 이 라이브러리의 API 설명서는 SageMaker Python SDK v2.199.0 문서의 분산 교육 API에 있습니다.