Repolode
aws

aws/sagemaker-python-sdk

Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 Python SDK로, 프레임워크와 내장 알고리즘, BYO 컨테이너를 지원합니다.

스타
2,270
포크
1,317
열린 이슈
237
마지막 커밋
어제

어떤 프로젝트인가요?

SageMaker Python SDK는 Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 오픈 소스 라이브러리입니다. 개발자는 PyTorch와 MXNet 같은 딥러닝 프레임워크, Amazon의 내장 알고리즘 또는 SageMaker 호환 Docker 컨테이너로 패키징한 사용자 지정 알고리즘을 사용해 Python 코드에서 학습 작업을 실행하고 모델을 호스팅할 수 있습니다.

3.x 버전은 학습용 ModelTrainer와 배포 및 추론용 ModelBuilder라는 통합 클래스를 중심으로 구성되며, 2.x의 프레임워크별 Estimator 및 Model 클래스를 대체합니다. 3.x는 core, training, serving, MLOps용 PyPI 패키지로 분리되어 있습니다. 모델 미세 조정 트레이너(SFT, DPO, RLAIF, RLVR)도 3.1 이상 버전에 포함되어 있습니다.

SDK는 Python 3.10~3.12를 사용하는 Unix/Linux와 macOS를 지원합니다. 사용하려면 AWS 계정과 SageMaker 권한이 있는 IAM 역할이 필요합니다.

핵심 기능

  • 01SageMaker에서 PyTorch, MXNet 및 기타 프레임워크로 모델을 학습합니다.
  • 02Amazon 내장 알고리즘을 사용하거나 자체 Docker 컨테이너를 가져와 사용할 수 있습니다.
  • 03ModelBuilder로 학습된 모델을 실시간 엔드포인트에 배포합니다.
  • 04LoRA 또는 전체 미세 조정을 사용해 파운데이션 모델을 미세 조정합니다.
  • 05core, training, serving, MLOps용 모듈식 패키지를 설치합니다.

활용 사례

  • Python에서 관리형 AWS GPU 인스턴스로 분산 학습 작업을 실행합니다.
  • 학습된 모델을 실시간 추론 엔드포인트로 배포합니다.
  • AWS에서 사용자 지정 데이터로 대규모 언어 모델을 미세 조정합니다.
  • 모델 레지스트리와 처리 작업을 활용해 MLOps 파이프라인을 구축합니다.

빠른 시작

  1. 1

    pip으로 PyPI에서 SDK를 설치합니다

    pip install sagemaker

사용 전 알아둘 점

  • Amazon SageMaker를 대상으로 하므로 AWS 계정과 IAM 권한이 필요합니다.
  • README에 따르면 Unix/Linux와 Mac만 지원합니다.
  • 버전 3.x에서는 버전 2의 Estimator, Model, Predictor 클래스가 제거되었습니다.

토픽

자주 묻는 질문

sagemaker-python-sdk은(는) 무엇인가요?
Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 Python SDK로, 프레임워크와 내장 알고리즘, BYO 컨테이너를 지원합니다. SageMaker Python SDK는 Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 오픈 소스 라이브러리입니다. 개발자는 PyTorch와 MXNet 같은 딥러닝 프레임워크, Amazon의 내장 알고리즘 또는 SageMaker 호환 Docker 컨테이너로 패키징한 사용자 지정 알고리즘을 사용해 Python 코드에서 학습 작업을 실행하고 모델을 호스팅할 수 있습니다.
sagemaker-python-sdk은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
sagemaker-python-sdk은(는) Apache-2.0 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
sagemaker-python-sdk은(는) 아직 유지보수되고 있나요?
활발히 유지보수 중. sagemaker-python-sdk의 마지막 커밋은 2026년 10월 23일입니다.
sagemaker-python-sdk은(는) 어디에 사용할 수 있나요?
Python에서 관리형 AWS GPU 인스턴스로 분산 학습 작업을 실행합니다. 학습된 모델을 실시간 추론 엔드포인트로 배포합니다. AWS에서 사용자 지정 데이터로 대규모 언어 모델을 미세 조정합니다. 모델 레지스트리와 처리 작업을 활용해 MLOps 파이프라인을 구축합니다.
sagemaker-python-sdk는 어떻게 설치하나요?
pip으로 PyPI에서 SDK를 설치합니다: pip install sagemaker
sagemaker-python-sdk은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/aws/sagemaker-python-sdk

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트