如何在 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, this is how you retrieve information about the environment: 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 以查看每个版本中安装的内容。