如何用Terraform在SageMaker模型包组中注册模型？

要在已创建的模型包组中注册模型，你需要使用aws_sagemaker_model_package资源——注意aws_sagemaker_model是用于部署的模型实体，并非模型包组内的注册项。

示例代码 #

# 已创建的模型包组（你已完成这一步，此处仅作关联参考）
resource "aws_sagemaker_model_package_group" "example" {
  model_package_group_name        = "example-model-group"
  model_package_group_description = "用于存放示例模型的包组"
}

# 在模型包组中注册模型
resource "aws_sagemaker_model_package" "example" {
  model_package_name              = "example-model-package-v1"
  model_package_group_name        = aws_sagemaker_model_package_group.example.name
  model_package_description       = "版本1的示例模型包"

  inference_specification {
    containers {
      image         = "763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.11.0-cpu"
      model_data_url = "s3://your-model-bucket/artifacts/model-v1.tar.gz"
    }

    supported_content_types        = ["application/json"]
    supported_response_mime_types  = ["application/json"]
  }

  approval_status = "Approved" # 可选：设置模型包审批状态，值包括Approved/Rejected/PendingManualApproval
}

关键参数说明 #

• model_package_group_name：通过引用已创建的模型包组资源，确保模型包与目标组关联
• inference_specification：定义模型的推理镜像、模型文件存储位置、支持的请求/响应格式等核心信息
• approval_status：控制模型包是否可被用于部署，根据业务需求选择对应状态

内容的提问来源于stack exchange，提问作者Karki