[英]How to get an Amazon ECR container URI for a specific model image in Sagemaker?
我想知道是否可以通过编程方式(使用 AWS CLI 或 Python)获取特定图像的 Amazon ECR 容器 URI。 例如,如果我需要 URL 作为eu-central-1
区域的最新linear-learner
(内置模型)图像。
预期结果:
664544806723.dkr.ecr.eu-central-1.amazonaws.com/linear-learner:latest
编辑:我找到了get_image_uri
的解决方案。 看起来这个 function 将被弃用,我不知道如何使用ImageURIProvider
来代替。
我们可以使用函数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
来代替。
较新版本的 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
从 sagemaker import image_uris container = sagemaker.image_uris.retrieve("linear-learner", boto3.Session().region_name)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.