Studio에서 Slurm 클러스터 설정

도메인을 생성하거나 도메인을 준비합니다. 도메인 생성에 대한 자세한 내용은 섹션을 참조하세요Amazon SageMaker AI 설정 가이드.

(선택 사항) 사용자 지정 FSx for Lustre 볼륨을 생성하여 도메인에 연결합니다.

1. FSx Lustre 파일 시스템이 의도한 도메인과 동일한 VPC에 존재하고 도메인에 있는 서브넷 중 하나에 있는지 확인합니다.

2. 의 지침을 따를 수 있습니다도메인에 사용자 지정 파일 시스템 추가.

(선택 사항) 보다 원활한 워크플로를 위해 클러스터에 태그를 추가하는 것이 좋습니다. 태그를 추가하는 방법에 대한 자세한 내용은 섹션을 참조하여 SageMaker AI 콘솔SageMaker HyperPod 클러스터 편집을 사용하여 클러스터를 업데이트하세요.

1. FSx for Lustre 파일 시스템에 Studio 도메인에 태그를 지정합니다. 이렇게 하면 Studio 스페이스를 시작하는 동안 파일 시스템을 식별하는 데 도움이 됩니다. 이렇게 하려면 클러스터에 다음 태그를 추가하여 FSx 파일 시스템 ID 로 식별합니다fs-id.

   태그 키 = “hyperpod-cluster-filesystem”, 태그 값 = “fs-id”.

2. Amazon Managed Grafana 워크스페이스를 Studio 도메인에 태그 지정합니다. 이는 Studio의 클러스터에서 직접 Grafana 워크스페이스에 빠르게 연결하는 데 사용됩니다. 이렇게 하려면 클러스터에 다음 태그를 추가하여 Grafana 워크스페이스 ID 로 식별합니다ws-id.

   태그 키 = “grafana-workspace”, 태그 값 = “ws-id”.

실행 역할에 다음 권한을 추가합니다.

SageMaker AI 실행 역할과 편집 방법에 대한 자세한 내용은 섹션을 참조하세요도메인 공간 권한 및 실행 역할 이해.

IAM 사용자 또는 그룹에 정책을 연결하는 방법을 알아보려면 IAM 자격 증명 권한 추가 및 제거를 참조하세요.

{
    "Version": "2012-10-17",            
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "ssm:StartSession",
                "ssm:TerminateSession"
            ],
            "Resource": "*"    
        },
        {
            "Effect": "Allow",
            "Action": [
                "sagemaker:CreateCluster",
                "sagemaker:ListClusters"
            ],
            "Resource": "*"    
        },
        {
            "Effect": "Allow",
            "Action": [
                "cloudwatch:PutMetricData",
                "cloudwatch:GetMetricData"
                ],
            "Resource": "*"
        },
        {
            "Effect": "Allow",
            "Action": [
                "sagemaker:DescribeCluster",
                "sagemaker:DescribeClusterNode",
                "sagemaker:ListClusterNodes",
                "sagemaker:UpdateCluster",
                "sagemaker:UpdateClusterSoftware"
            ],
            "Resource": "arn:aws:sagemaker:region:account-id:cluster/*"    
        }
    ]
}

태그 키 = “SSMSessionRunAs” 및 태그 값 = “os user”를 사용하여이 IAM 역할에 태그를 추가합니다. os user 여기서는 Slurm 클러스터에 대해 설정한 것과 동일한 사용자입니다. AWS Systems Manager 에이전트(SSM 에이전트)의 Run As 기능을 사용하여 IAM 역할 또는 사용자 수준에서 SageMakerSageMaker HyperPod 클러스터에 대한 액세스를 관리합니다. 이 기능을 사용하면 IAM 역할 또는 사용자와 연결된 운영 체제(OS) 사용자를 사용하여 각 SSM 세션을 시작할 수 있습니다.

실행 역할에 태그를 추가하는 방법에 대한 자세한 내용은 IAM 역할 태그 지정을 참조하세요.

Linux 및 macOS 관리형 노드에 대한 Run As 지원을 켭니다. Run As 설정은 계정 전체이며 모든 SSM 세션이 성공적으로 시작되려면 필요합니다.

(선택 사항) Studio for Slurm 클러스터에서 작업 보기 제한. Studio에서 볼 수 있는 작업에 대한 자세한 내용은 섹션을 참조하세요업무.