如何在 Sagemaker 中获取特定模型映像的 Amazon ECR 容器 URI？

Question

我想知道是否可以通过编程方式（使用 AWS CLI 或 Python）获取特定图像的 Amazon ECR 容器 URI。例如，如果我需要linear-learner该地区最新（内置模型）图像的 URL eu-central-1。

预期结果：

664544806723.dkr.ecr.eu-central-1.amazonaws.com/linear-learner:latest

编辑：我已经找到了解决方案get_image_uri。看来这个功能将被弃用，我不知道如何使用ImageURIProvider。

Answer 1

较新版本的 SageMaker SDK 有一个更集中的 API 用于获取 URI：

import sagemaker 
sagemaker.image_uris.retrieve("linear-learner", "eu-central-1")

这给出了预期的结果：

664544806723.dkr.ecr.eu-central-1.amazonaws.com/linear-learner:1