创建自定义项目模板 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

创建自定义项目模板

如果 SageMaker 提供的模板不能满足您的需求(例如,您希望在 CodePipeline 中使用多个阶段或自定义批准步骤进行更复杂的编排),请创建自己的模板。

我们建议首先使用 SageMaker 提供的模板来了解如何整理代码和资源,并在此基础上进行构建。为此,启用 SageMaker 模板的管理员访问权限后,登录 https://console.aws.amazon.com/servicecatalog/,选择产品组合,然后选择已导入。有关 Service Catalog 的信息,请参阅《Service Catalog 用户指南》中的 Service Catalog 概述

创建自己的项目模板以自定义您的 MLOps 项目。SageMaker 项目模板是 Service Catalog 预置产品,用于为您的 MLOps 项目预置资源。

要创建自定义项目模板,请完成以下步骤。

  1. 创建产品组合。有关信息,请参阅步骤 3:创建 Service Catalog 产品组合

  2. 创建产品。产品是 CloudFormation 模板。您可以创建多个版本的产品。有关信息,请参阅步骤 4:创建 Service Catalog 产品

    要使产品与 SageMaker 项目配合使用,请向产品模板添加以下参数。

    SageMakerProjectName: Type: String Description: Name of the project SageMakerProjectId: Type: String Description: Service generated Id of the project.
    重要

    我们建议您将 CodeCommit 存储库封装到 SageMaker 代码存储库中,这样,项目的存储库就能在 VPC 模式下可见。示例模板和所需的附加内容如以下代码示例所示。

    原始(示例)模板:

    ModelBuildCodeCommitRepository: Type: AWS::CodeCommit::Repository Properties: # Max allowed length: 100 chars RepositoryName: !Sub sagemaker-${SageMakerProjectName}-${SageMakerProjectId}-modelbuild # max: 10+33+15+10=68 RepositoryDescription: !Sub SageMaker Model building workflow infrastructure as code for the Project ${SageMakerProjectName} Code: S3: Bucket: SEEDCODE_BUCKETNAME Key: toolchain/model-building-workflow-v1.0.zip BranchName: main

    要在 VPC 模式下添加的附加内容:

    SageMakerRepository: Type: AWS::SageMaker::CodeRepository Properties: GitConfig: RepositoryUrl: !GetAtt ModelBuildCodeCommitRepository.CloneUrlHttp Branch: main
  3. 添加启动约束。启动约束指定 IAM 角色,用户启动产品时 Service Catalog 将代入此角色。有关信息,请参阅步骤 6:添加启动约束以分配 IAM 角色

  4. https://console.aws.amazon.com/servicecatalog/ 上预置产品以测试模板。如果您对模板感到满意,请继续执行下一步以使该模板在 Studio 中可用。

  5. 将您在步骤 1 中创建的 Service Catalog 产品组合的访问权限授予您的 Studio 执行角色。使用 Studio 域执行角色或具有 Studio 访问权限的用户角色。有关向产品组合添加角色的信息,请参阅步骤 7:向最终用户授予产品组合的访问权限

  6. 要使您的项目模板在 Studio 的组织模板列表中可用,请为在步骤 2 中创建的 Service Catalog 产品创建一个具有以下键和值的标签。

    • sagemaker:studio-visibility

    • true

完成这些步骤后,组织中的 Studio 用户就可以使用您创建的模板来创建项目,方法是按照使用 Amazon SageMaker Studio 创建 MLOps 项目中的步骤进行操作,并在选择模板时选择组织模板