» aws_sagemaker_model

Provides a SageMaker model resource.

» Example Usage

Basic usage:

resource "aws_sagemaker_model" "m" {
    name = "my-model"
    execution_role_arn = "${aws_iam_role.foo.arn}"

    primary_container {
        image = "174872318107.dkr.ecr.us-west-2.amazonaws.com/kmeans:1"

resource "aws_iam_role" "r" {
  assume_role_policy = "${data.aws_iam_policy_document.assume_role.json}"

data "aws_iam_policy_document" "assume_role" {
  statement {
    actions = [ "sts:AssumeRole" ]
    principals {
      type = "Service"
      identifiers = [ "sagemaker.amazonaws.com" ]

» Argument Reference

The following arguments are supported:

  • name - (Optional) The name of the model (must be unique). If omitted, Terraform will assign a random, unique name.
  • primary_container - (Optional) The primary docker image containing inference code that is used when the model is deployed for predictions. If not specified, the container argument is required. Fields are documented below.
  • execution_role_arn - (Required) A role that SageMaker can assume to access model artifacts and docker images for deployment.
  • container (Optional) - Specifies containers in the inference pipeline. If not specified, the primary_container argument is required. Fields are documented below.
  • enable_network_isolation (Optional) - Isolates the model container. No inbound or outbound network calls can be made to or from the model container.
  • vpc_config (Optional) - Specifies the VPC that you want your model to connect to. VpcConfig is used in hosting services and in batch transform.
  • tags - (Optional) A mapping of tags to assign to the resource.

The primary_container and container block both support:

  • image - (Required) The registry path where the inference code image is stored in Amazon ECR.
  • model_data_url - (Optional) The URL for the S3 location where model artifacts are stored.
  • container_hostname - (Optional) The DNS host name for the container.
  • environment - (Optional) Environment variables for the Docker container. A list of key value pairs.

» Attributes Reference

The following attributes are exported:

  • name - The name of the model.
  • arn - The Amazon Resource Name (ARN) assigned by AWS to this model.

» Import

Models can be imported using the name, e.g.

$ terraform import aws_sagemaker_model.test_model model-foo