Amazon SageMaker 部署 AIGC 应用:模型训练、优化与 Web 前端集成
背景
![图片]
Amazon SageMaker 是亚马逊云科技全托管机器学习服务,提供数据处理、模型训练、超参数优化及部署推理能力。支持开发者使用 Notebook 实例调试 AIGC 模型,借助分布式训练集群加速迭代,并将模型高效部署为可自动扩缩容的 Inference Endpoint。深度集成 S3 存储、CloudWatch 监控等,以容器化架构和 Kubernetes 集群管理实现从开发到生产的无缝衔接。
前言
本文将通过 Amazon SageMaker Notebook 实例完成 AIGC 模型的测试与验证,再将模型部署至 Amazon SageMaker Inference Endpoint 实现服务化,最后利用 Amazon Cloud 9 调试 Web 应用,并调用已部署的 SageMaker Inference Endpoint 实现 AIGC 模型的实际应用。
Amazon SageMaker 部署 AIGC 模型
步骤一:创建亚马逊云科技账户
- 注册亚马逊云科技账号。
- 进入注册页面,输入电子邮件地址,编辑账户名称,验证电子邮件地址。
- 设置用户信息,输入根用户密码,选择个人或企业。
- 添加付款信息,输入有效的支付方式。
- 用户信息验证,选择接收验证码的联系方式。
- 客户验证与激活账户,选择支持服务计划。
- 注册成功。
步骤二:创建 Amazon SageMaker Notebook 实例
- 登录亚马逊云科技控制台。
- 进入 SageMaker 服务,搜索 SageMaker。
- 进入 Notebook 笔记本实例控制面板(左侧菜单栏,点击笔记本实例)。
- 创建笔记本实例。
- 配置笔记本实例设置,设置名称、选择实例类型及存储大小。
- 配置权限,创建 IAM 角色用于调用 SageMaker 和 S3 等服务。
- 默认配置 IAM 角色。
- 检查配置信息,确认无误后创建笔记本实例。
- 等待实例创建成功,进入 Jupyter Notebook。
- 构建 Stable-Diffusion 环境(Jupyter Lab 控制台,选择 Terminal)。
- 下载 Notebook 代码文件。
- 打开 Notebook 文件,内核选择 conda_pytorch_p39。
步骤三:Amazon Cloud 9 构建前后端 Web 应用
- 亚马逊云科技控制台进入 Cloud 9 服务。
- 创建 Cloud 9 环境,配置名字,其他保持默认。
- 进入 Amazon Cloud 9 环境。
- 在控制台中输入指令下载保存 Web 应用程序代码压缩文件并解压。 SampleWebApp 文件夹包含以下内容:后端代码 app.py(接收前端请求并调用 SageMaker Endpoint 将文字生成图片),两个前端 html 文件 image.html 和 index.html。
- Amazon Cloud 9 上安装 Flask 和 boto3。
- 运行 app.py,点击 Run 按钮运行代码。
- 预览 Web 前端页面。
- 在 Amazon Cloud 9 环境打开 Web 页面。
- 输入信息生成图片(自定义图片长度和宽度,Prompt 输入框处输入语句或描绘性词汇生成图片)。详细描述需求信息以便 AI 绘制出的图片更贴合预期,例如通过描述性词语如 cartoon style、van gogh style、realistic 表明图片的整体风格。可将长度和宽度都设置成 512,输入描绘性语句生成卡通风格的宇航员猴子图片以及超级英雄的图片。
- 成功页面展示。
Amazon SageMaker 工作原理
![图片]
Amazon SageMaker 全托管机器学习服务,整合数据处理、模型训练、超参数优化、部署推理的全流程能力。


