aws/sagemaker-python-sdk

これは何？

主な機能

• 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