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

0

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

1 Antwort
0

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

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

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

profile picture
EXPERTE
beantwortet vor 8 Monaten

Du bist nicht angemeldet. Anmelden um eine Antwort zu veröffentlichen.

Eine gute Antwort beantwortet die Frage klar, gibt konstruktives Feedback und fördert die berufliche Weiterentwicklung des Fragenstellers.

Richtlinien für die Beantwortung von Fragen