行业资讯

亚马逊SageMaker SDK v3更新:自带模型工作流进一步简化

🎧 本文 AI 播客

优秘智能观察:云原生AI开发工具的迭代正持续降低企业AI落地的技术门槛,亚马逊最新发布的SageMaker Python SDK v3脚本模式更新,为各类企业的模型训练与部署效率升级提供了新的可行路径。

SageMaker SDK v3核心升级:简化自带模型工作流

早在2021年,亚马逊就推出了SageMaker脚本模式,支持开发者基于AWS托管框架容器编写自定义训练与推理代码,无需自行构建维护Docker镜像即可运行自定义算法。本次v3版本SDK为全新重构设计,进一步优化了自带模型的全流程体验:

  • 取消框架专属估算器类(SKLearn、PyTorch、XGBoost等),用统一的ModelTrainer类处理训练任务,ModelBuilder类处理部署任务
  • 新增SourceCode配置对象,运行时自动同步本地源代码目录到训练任务中,支持自定义容器、AWS深度学习容器或第三方容器,代码运行时自动注入无需打包进镜像

本次升级带来三大核心价值:

  • 更快迭代效率:修改训练脚本后可直接重新运行,无需重构容器
  • 全容器控制权:支持开发者自行在镜像中安装系统包或CUDA库,SDK不限制镜像内部环境
  • 跨框架统一API:无论使用scikit-learn、PyTorch、Stable Diffusion还是自定义C++推理二进制文件,操作接口完全一致

两大核心应用场景示例

本次官方更新同步提供了两类典型AI开发场景的完整实操案例,全程仅需使用ModelTrainer与ModelBuilder两个核心类即可完成全流程操作:

  • 表格类机器学习场景:基于糖尿病数据集训练scikit-learn随机森林模型,使用高性能模型服务器Deep Java Library (DJL) Serving部署为实时接口
  • 生成式AI微调场景:使用Hugging Face Accelerate实现多GPU分布式训练,对Stable Diffusion 3.5进行LoRA微调

核心逻辑为通过SourceCode对象绑定本地代码目录与运行指令/入口脚本,任务启动时SageMaker自动同步目录到容器内运行,无需将代码打包进镜像,完整示例代码可参考官方GitHub仓库。

从v2到v3的架构变化对比

  • 训练类:v2为各框架专属Estimator,v3统一为ModelTrainer单类
  • 部署类:v2为Model+Predictor组合,v3通过ModelBuilder部署接口,预测直接通过invoke()处理
  • 容器支持:v2仅支持AWS托管框架镜像,v3支持自定义镜像、AWS DLC镜像、第三方镜像
  • 代码注入:v2为框架专属的entry_point+source_dir配置,v3为统一的SourceCode对象配置

落地前置要求

企业如需试用该新特性,需提前满足以下条件:

  • 开通AWS账号并获得Amazon SageMaker AI访问权限
  • IAM执行角色具备SageMaker与S3存储相关权限
  • 安装SageMaker Python SDK v3(版本≥3.0)
  • 训练容器镜像已推送至Amazon ECR
  • 配置用于存储训练数据与模型产物的S3存储桶
  • 可选配置:开通SageMaker MLflow应用用于实验跟踪,如使用SageMaker Studio的JupyterLab空间运行需提前开启域级Docker访问权限

优秘智能点评:对于正在推进AI化转型的企业而言,云侧AI开发工具的迭代意味着更低的技术试错成本与更快的模型落地效率。无论是1人创业团队还是中大型企业,都可以借助这类工具简化AI模型的训练部署流程,将更多精力放在业务场景适配而非底层基建维护上。如需了解优秘智能基于成熟AI底座为企业提供的营销智脑、数字分身等开箱即用的AI产品方案,可访问官网umi6.com咨询详情。

读完这篇,下一步可以这样落地

把文章里的方法,拆成你的企业 AI 化方案

留下联系方式,顾问会结合你的行业、团队规模和当前业务目标,给你一份更具体的落地建议。

先做 AI 测评
结合行业场景输出落地路径信息严格保密
内容不是终点

让顾问帮你判断,哪些 AI 场景值得先做

一次沟通即可梳理当前业务里最容易产生回报的 AI 应用入口。

先做 AI 测评
结合行业场景输出落地路径信息严格保密