如何在SageMaker培训作业脚本中获取当前的作业名称？

Question

我在AWS-SageMaker框架上写了一些培训工作。

对于某些要求，它需要知道当前运行的作业的名称。

我知道这段代码适用于它...

import sagemaker_containers
env = sagemaker_containers.training_env()
job_name = env['job_name']

但是sagemaker_containers软件包已被弃用。 （我在it's GitHub上看到了）

我该怎么办？

我上个月刚开始学习这个平台。我将不胜感激任何建议。谢谢。

Answer 1

对于使用不建议使用的sagemaker_containers的旧容器，您描述的方法是正确的。

对于使用sagemaker-training-toolkit的较新容器，这是您检索有关环境的信息的方式：https://github.com/aws/sagemaker-training-toolkit#get-information-about-the-container-environment

from sagemaker_training import environment

env = environment.Environment()

job_name = env["job_name"]

您可以检查DLC Release Notes来查看每个版本中安装了什么。