File konfigurasi

Amazon SageMaker Python SDK mendukung pengaturan nilai default untuk tipe primitif AWS infrastruktur. Setelah administrator mengkonfigurasi default ini, mereka secara otomatis diteruskan ketika panggilan Python SageMaker didukung. SDK APIs Argumen untuk fungsi dekorator dapat dimasukkan ke dalam file konfigurasi. Ini agar Anda dapat memisahkan pengaturan yang terkait dengan infrastruktur dari basis kode. Untuk informasi selengkapnya tentang parameter dan argumen untuk fungsi dan metode jarak jauh, lihat Kelas fungsi jarak jauh dan spesifikasi metode.

Anda dapat mengatur pengaturan infrastruktur untuk konfigurasi jaringan, IAM peran, folder Amazon S3 untuk input, data output, dan tag di dalam file konfigurasi. File konfigurasi dapat digunakan saat menjalankan fungsi menggunakan dekorator @remote atau. RemoteExecutor API

Contoh file konfigurasi yang mendefinisikan dependensi, sumber daya, dan argumen lainnya berikut. Contoh file konfigurasi ini digunakan untuk memanggil fungsi yang dimulai baik menggunakan dekorator @remote atau. 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'

Dekorator @remote dan RemoteExecutor akan mencari Dependencies dalam file konfigurasi berikut:

Lokasi default untuk file konfigurasi ini bergantung pada, dan relatif terhadap, lingkungan Anda. Contoh kode berikut mengembalikan lokasi default admin dan file konfigurasi pengguna Anda. Perintah ini harus dijalankan di lingkungan yang sama di mana Anda menggunakan SageMaker PythonSDK.

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"))

Anda dapat mengganti lokasi default file ini dengan menyetel variabel SAGEMAKER_ADMIN_CONFIG_OVERRIDE dan SAGEMAKER_USER_CONFIG_OVERRIDE lingkungan untuk jalur file konfigurasi yang ditentukan admin dan yang ditentukan pengguna, masing-masing.

Jika kunci ada di file konfigurasi yang ditentukan admin dan yang ditentukan pengguna, nilai dalam file yang ditentukan pengguna akan digunakan.