구성 파일

Amazon SageMaker Python은 AWS 인프라 기본 유형에 대한 기본값 설정을 SDK 지원합니다. 관리자가 이러한 기본값을 구성하면 SageMaker Python이 를 SDK 지원하면 자동으로 전달됩니다APIs. 데코레이터 함수의 인수는 구성 파일 내에 넣을 수 있습니다. 이를 통해 인프라와 관련된 설정을 코드 베이스에서 분리할 수 있습니다. 원격 함수 및 메서드의 파라미터와 인수에 대한 자세한 내용은 원격 함수 클래스 및 메서드 사양을 참조하세요.

네트워크 구성, IAM 역할, Amazon S3 폴더의 입력, 출력 데이터 및 구성 파일 내 태그에 대한 인프라 설정을 설정할 수 있습니다. 구성 파일은 @remote 데코레이터 또는 RemoteExecutor 를 사용하여 함수를 호출할 때 사용할 수 있습니다API.

다음은 종속성, 리소스 및 기타 인수를 정의하는 예제 구성 파일입니다. 이 예제 구성 파일은 @remote 데코레이터 또는 를 사용하여 시작된 함수를 호출하는 데 사용됩니다 RemoteExecutor API.

SchemaVersion: '1.0'
SageMaker:
  PythonSDK:
    Modules:
      RemoteFunction:
        Dependencies: 'path/to/requirements.txt'
        EnableInterContainerTrafficEncryption: true
        EnvironmentVariables: {'EnvVarKey': 'EnvVarValue'}
        ImageUri: '366666666666.dkr.ecr.us-west-2.amazonaws.com/my-image:latest'
        IncludeLocalWorkDir: true
        CustomFileFilter: 
          IgnoreNamePatterns:
          - "*.ipynb"
          - "data"
        InstanceType: 'ml.m5.large'
        JobCondaEnvironment: 'your_conda_env'
        PreExecutionCommands:
            - 'command_1'
            - 'command_2'
        PreExecutionScript: 'path/to/script.sh'
        RoleArn: 'arn:aws:iam::366666666666:role/MyRole'
        S3KmsKeyId: 'yourkmskeyid'
        S3RootUri: 's3://amzn-s3-demo-bucket/my-project'
        VpcConfig:
            SecurityGroupIds: 
            - 'sg123'
            Subnets: 
            - 'subnet-1234'
        Tags: [{'Key': 'yourTagKey', 'Value':'yourTagValue'}]
        VolumeKmsKeyId: 'yourkmskeyid'

@remote 데코레이터 및 RemoteExecutor가 다음 구성 파일에서 Dependencies을 검색합니다.

이러한 구성 파일의 기본 위치는 환경에 따라 다르고 환경과 관련이 있습니다. 다음 코드 예제에서는 관리자 및 사용자 구성 파일의 기본 위치를 반환합니다. 이러한 명령은 SageMaker Python 를 사용하는 것과 동일한 환경에서 실행되어야 합니다SDK.

import os
from platformdirs import site_config_dir, user_config_dir

#Prints the location of the admin config file
print(os.path.join(site_config_dir("sagemaker"), "config.yaml"))

#Prints the location of the user config file
print(os.path.join(user_config_dir("sagemaker"), "config.yaml"))

관리자 정의 구성 파일 및 사용자 정의 구성 파일 경로에 각각 SAGEMAKER_ADMIN_CONFIG_OVERRIDE 및 SAGEMAKER_USER_CONFIG_OVERRIDE 환경 변수를 설정하여 이러한 파일의 기본 위치를 재정의할 수 있습니다.

관리자 정의 구성 파일 및 사용자 정의 구성 파일 모두에 키가 있는 경우 사용자 정의 파일의 값이 사용됩니다.