如何在python脚本中返回AWS Sagemaker笔记本实例的名称?

时间:2020-01-28 19:51:20

标签: python amazon-web-services amazon-sagemaker

有点案例,基本上,我正在尝试编写一个通用脚本,该脚本可以在AWS Sagemaker上设置为cronjob,如果实例仍处于活动状态,它将通知用户。 如果可能,我希望能够在cronjob中调用AWS Sagemaker实例的名称(与活动的jupyter笔记本的名称不同)。

在boto3或sagemaker文档中找不到与此相关的任何内容。我知道实例名称包含在URL路径中,但是类似地,还没有找到引用它的方法。我希望这不是预期的功能,因此可能会继续在配置文件中手动设置值,但是如果有人有任何创造性的解决方案,我将不胜感激!

1 个答案:

答案 0 :(得分:1)

这是一个如何获取实例名称的例子

def get_notebook_name():
    log_path = '/opt/ml/metadata/resource-metadata.json'
    with open(log_path, 'r') as logs:
        _logs = json.load(logs)
    return _logs['ResourceName']

发件人:

https://github.com/aws-samples/amazon-sagemaker-notebook-instance-lifecycle-config-samples/blob/62c44aa5e69f4266955476f24647b99d9b597aaf/scripts/auto-stop-idle/autostop.py#L79