时间戳

首页 > 解决方案 > 如何在 Sagemaker 中获取特定模型映像的 Amazon ECR 容器 URI？

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

问题描述

我想知道是否可以通过编程方式（使用 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。

标签： pythonamazon-web-servicesaws-cliamazon-sagemakeramazon-ecr

解决方案

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

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

这给出了预期的结果：

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

推荐阅读