精选的开源库清单,涵盖生产环境机器学习的部署、监控、版本管理、扩展与安全。
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
通过 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 数据每日更新。
相关项目
一组基于 Amazon SageMaker 的示例 Jupyter 笔记本,演示如何构建、训练和部署机器学习模型。
Hugging Face 推出的 Python 库,一行代码即可加载 Hub 上的公开 AI 数据集,并用基于 Arrow 的工具高效预处理。
用于跟踪、可视化和管理机器学习实验、数据集与模型的 Python 库及托管平台。
开源的 AI 训练跟踪与可视化平台,记录指标、超参数与硬件状态,支持云端或私有化部署。
开源 Python 平台,把机器学习与 AI 工作流组织成可在本地或云端基础设施上运行的流水线,涵盖模型训练到智能体。