設定 的IAM許可 MLflow

您必須設定必要的IAM服務角色，才能在 Amazon MLflow中開始使用 SageMaker。

如果您建立新的 Amazon SageMaker 網域以在 Studio 中存取實驗，您可以在網域設定期間設定必要的IAM許可。如需詳細資訊，請參閱建立新網域時設定MLflowIAM許可。

若要使用IAM主控台設定許可，請參閱 在IAM主控台中建立必要的IAM服務角色。

您必須設定sagemaker-mlflow動作的授權控制。您可以選擇性地定義更精細的授權控制，以管理動作特定的MLflow許可。如需詳細資訊，請參閱建立動作特定的授權控制項。

建立新網域時設定MLflowIAM許可

為組織設定新的 Amazon SageMaker 網域時，您可以透過使用者和 ML 活動設定來設定網域服務角色的IAM許可。

下列 MLflow ML 活動可在 Amazon SageMaker Role Manager 中使用：

如需 Role Manager 中 ML 活動的詳細資訊，請參閱 機器學習 (ML) 活動參考。

在IAM主控台中建立必要的IAM服務角色

如果您未建立或更新網域服務角色，則必須改為在IAM主控台中建立下列服務角色，才能建立和使用MLflow追蹤伺服器：

IAM 追蹤伺服器IAM服務角色的政策

追蹤伺服器會使用追蹤伺服器IAM服務角色來存取所需的資源，例如 Amazon S3 和 SageMaker 模型登錄檔。

建立追蹤伺服器IAM服務角色時，請使用下列IAM信任政策：

{
     "Version": "2012-10-17",
     "Statement": [
         {
             "Effect": "Allow",
             "Principal": {
                 "Service": [                     
                      "sagemaker.amazonaws.com"
                 ]
             },
             "Action": "sts:AssumeRole"
         }
     ]
 }

在IAM主控台中，將下列許可政策新增至您的追蹤伺服器服務角色：

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "s3:Get*",
                "s3:Put*",
                "s3:List*",
                "sagemaker:AddTags",
                "sagemaker:CreateModelPackageGroup",
                "sagemaker:CreateModelPackage",
                "sagemaker:UpdateModelPackage",
                "sagemaker:DescribeModelPackageGroup"
            ],
            "Resource": "*"
        }
    ]
}

IAM SageMaker IAM服務角色的政策

SageMaker 服務角色由存取MLflow追蹤伺服器的用戶端使用，且需要呼叫 MLflow REST 的許可APIs。 SageMaker 服務角色也需要 SageMaker API許可才能建立、更新、啟動、停止和刪除追蹤伺服器。

您可以建立新的角色或更新現有角色。 SageMaker 服務角色需要下列政策：

{
    "Version": "2012-10-17",    
    "Statement": [        
        {            
            "Effect": "Allow",            
            "Action": [
                "sagemaker-mlflow:*",
                "sagemaker:CreateMlflowTrackingServer",
                "sagemaker:UpdateMlflowTrackingServer",
                "sagemaker:DeleteMlflowTrackingServer",
                "sagemaker:StartMlflowTrackingServer",
                "sagemaker:StopMlflowTrackingServer",
                "sagemaker:CreatePresignedMlflowTrackingServerUrl"
            ],            
            "Resource": "*"        
        }        
    ]
}

建立動作特定的授權控制項

您必須為 設定授權控制sagemaker-mlflow，並且可以選擇設定動作特定的授權控制，以管理使用者在MLflow追蹤伺服器上擁有的更精細MLflow許可。

下列命令會建立名為 的檔案mlflow-policy.json，為您的追蹤伺服器提供所有可用 SageMaker MLflow動作的IAM許可。您可以選擇性地限制使用者擁有的許可，方法是選擇您希望該使用者執行的特定動作。如需可用的動作清單，請參閱 IAM 支援 的動作 MLflow。

# Replace "Resource":"*" with "Resource":"TrackingServerArn" 
# Replace "sagemaker-mlflow:*" with specific actions

printf '{
    "Version": "2012-10-17",    
    "Statement": [        
        {            
            "Effect": "Allow",            
            "Action": "sagemaker-mlflow:*",            
            "Resource": "*"        
        }        
    ]
}' > mlflow-policy.json

使用 mlflow-policy.json 檔案來建立使用 IAM的政策 AWS CLI。

aws iam create-policy \
  --policy-name MLflowPolicy \
  --policy-document file://mlflow-policy.json

擷取您的帳戶 ID 並將政策連接至您的IAM角色。

# Get your account ID
aws sts get-caller-identity

# Attach the IAM policy using your exported role and account ID
aws iam attach-role-policy \
  --role-name $role_name \
  --policy-arn arn:aws:iam::123456789012:policy/MLflowPolicy

IAM 支援 的動作 MLflow

授權存取控制支援下列 SageMaker MLflow動作：