python - 如何在 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

我们可以使用函数get_image_uri :

from sagemaker.amazon.amazon_estimator import get_image_uri

region = boto3.Session().region_name
#or region = 'eu-central-1'

get_image_uri(region_name=region,
              repo_name='linear-learner',
              repo_version='latest')

输出:

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

警告:

'get_image_uri' method will be deprecated in favor of 'ImageURIProvider' class in SageMaker Python SDK v2.

看起来这个功能将被弃用，我找不到如何使用 ImageURIProvider反而。