我如何使用Python SDK将环境变量传递给亚马逊萨吉式制造商的自定义培训脚本？ 我在Amazon Sagemaker中使用脚本训练自定义模型，并使用Python SDK启动工作。我想将一些环境变量（例如API键或配置标志）传递到培训j ...

Question

在我的培训脚本中，我尝试读取变量：

import os

api_key = os.environ.get('MY_API_KEY')
print("API Key:", api_key)

这是使用Python SDK将环境变量传递给SageMaker培训工作的正确方法？我应该知道任何局限性或最佳实践，尤其是对于敏感信息（例如API键）？ 是的，您的方法是正确的。在估计器中使用环境参数并在脚本中使用

os.environ.get()

访问变量是传递萨格人中环境变量的标准方法。正如@furas在他们的

comment

中指出的那样，

os.environ.get()

是Python中的常见方法。

Answer 1

import boto3  
import os  

secret_name = os.environ.get('API_KEY_SECRET_NAME')  
region = os.environ.get('AWS_REGION', 'us-west-2')  

client = boto3.client('secretsmanager', region_name=region)  
secret_value = client.get_secret_value(SecretId=secret_name)  
api_key = secret_value['SecretString']

print("API Key:", api_key)

这将实际的秘密置于您的环境配置之外，并允许通过IAM更好地访问控制。