Repolode
aws

aws/sagemaker-python-sdk

用于在 Amazon SageMaker 上训练和部署机器学习模型的 Python SDK。

Star
2,270
Fork
1,317
未关闭 Issue
237
最近提交
昨天

这是什么?

SageMaker Python SDK 是一个开源库,用于在 Amazon SageMaker 上训练和部署机器学习模型。开发者可以直接用 Python 代码提交训练任务、托管模型,支持 PyTorch、MXNet 等深度学习框架,也支持 Amazon 内置算法以及打包在 SageMaker 兼容 Docker 容器中的自定义算法。

3.x 版本以统一的类为核心:ModelTrainer 负责训练,ModelBuilder 负责部署和推理,取代了 2.x 中按框架划分的 Estimator 和 Model 类。3.x 被拆分为 core、train、serve、mlops 等独立 PyPI 包,3.1 及以后版本还提供 SFT、DPO、RLAIF、RLVR 等基础模型微调训练器。

该 SDK 支持 Linux/Unix 和 macOS,测试于 Python 3.10 至 3.12。使用前需要 AWS 账户,并配置具有 SageMaker 相关权限的 IAM 角色。

核心功能

  • 01在 SageMaker 上使用 PyTorch、MXNet 等框架训练模型。
  • 02可使用 Amazon 内置算法,也可自带 Docker 容器。
  • 03通过 ModelBuilder 将训练好的模型部署为实时推理端点。
  • 04支持使用 LoRA 或全量微调对基础模型进行微调。
  • 05提供按 core、训练、推理和 MLOps 拆分的模块化 PyPI 包。

适用场景

  • 在 AWS 托管的 GPU 实例上从 Python 运行分布式训练任务。
  • 将训练好的模型部署为实时推理端点。
  • 在 AWS 上用自有数据微调大语言模型。
  • 结合模型注册表和处理任务构建 MLOps 流水线。

快速上手

  1. 1

    通过 pip 从 PyPI 安装 SDK

    pip install sagemaker

使用前须知

  • 该 SDK 面向 Amazon SageMaker,需要 AWS 账户和 IAM 权限。
  • README 称仅支持 Unix/Linux 和 Mac 系统。
  • 3.x 版本不再支持 Estimator、Model、Predictor 等 2.x 类。

标签

常见问题

sagemaker-python-sdk 是什么?
用于在 Amazon SageMaker 上训练和部署机器学习模型的 Python SDK。 SageMaker Python SDK 是一个开源库,用于在 Amazon SageMaker 上训练和部署机器学习模型。开发者可以直接用 Python 代码提交训练任务、托管模型,支持 PyTorch、MXNet 等深度学习框架,也支持 Amazon 内置算法以及打包在 SageMaker 兼容 Docker 容器中的自定义算法。
sagemaker-python-sdk 可以免费使用吗?用的什么许可证?
sagemaker-python-sdk 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
sagemaker-python-sdk 还在维护吗?
活跃维护中。sagemaker-python-sdk 最近一次提交是在 2026年10月23日。
sagemaker-python-sdk 适合用来做什么?
在 AWS 托管的 GPU 实例上从 Python 运行分布式训练任务。 将训练好的模型部署为实时推理端点。 在 AWS 上用自有数据微调大语言模型。 结合模型注册表和处理任务构建 MLOps 流水线。
sagemaker-python-sdk 怎么安装和运行?
通过 pip 从 PyPI 安装 SDK: pip install sagemaker
在哪里获取 sagemaker-python-sdk?
源代码和安装说明在 GitHub 上:https://github.com/aws/sagemaker-python-sdk

概要由 AI 根据项目 README 生成;GitHub 数据每日更新。

相关项目