如何将训练好的模型导出为容器镜像到Amazon ECR?

0

【以下的问题经过翻译处理】 我想在Sagemaker Studio中训练和构建模型,然后能够将模型导出为容器镜像到ECR,这样我就可以通过将ECR镜像共享到另一个账户,在外部平台使用该模型,从ECR镜像创建容器。

1개 답변
0

【以下的回答经过翻译处理】 在SageMaker中训练好的模型以.tar.gz格式储存在S3中,您可以使用它们来部署到端点,甚至在本地测试(从tar文件中提取模型文件)。 如果您正在使用内置算法,可以直接将.tar.gz文件分享给第二个帐户,并在第二个帐户中部署模型,因为内置算法容器可以从任何AWS帐户访问。

如果您正在使用自定义的训练镜像(文档在此),您可以将该镜像推送到ECR,并允许第二个帐户拉取镜像,然后将该镜像与已训练的模型一起使用。但需要注意的是,此时Studio不支持直接构建Docker镜像,您可以使用SageMaker Notebook Instances

建议将模型(.tar.gz)和镜像(Docker)保持分开,这样可以轻松地重新训练和部署新版本的模型,而无需每次更新镜像。

profile picture
전문가
답변함 8달 전

로그인하지 않았습니다. 로그인해야 답변을 게시할 수 있습니다.

좋은 답변은 질문에 명확하게 답하고 건설적인 피드백을 제공하며 질문자의 전문적인 성장을 장려합니다.

질문 답변하기에 대한 가이드라인