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



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

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

下面是如何获取实例名称的示例

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

最新更新