Sagemaker Tensorflow 2.0 端点

Sagemaker Tensorflow 2.0 endpoint

我有一个 tensorflow 2.0 模型,我想将其部署到 AWS sagemaker 端点。我已将模型移至 S3 存储桶并执行以下代码,但由于没有 TF 2.0 图像而出现以下错误。如果我尝试使用不同的版本(例如 1.4、1.8)进行部署,我会收到 ping 超时错误。

是否可以轻松创建一个?我找不到好的教程可以遵循。或者亚马逊会在某个时候部署一个。

失败。原因:图像'520713654638.dkr.ecr.eu-west-1.amazonaws.com/sagemaker-tensorflow:2.0-cpu-py2'不存在..

    from sagemaker.tensorflow.model import TensorFlowModel
    sagemaker_model = TensorFlowModel(model_data = 's3://sagemaker-eu-west-1- 
    273649867642/model/model.tar.gz',
                                  role = role,
                                  framework_version = '2.0',
                                  entry_point = 'train.py')

    %%time
    predictor = sagemaker_model.deploy(initial_instance_count=1,
                                   instance_type='ml.m4.xlarge')

似乎也没有图像支持 python 3,尽管它们建议您在设置模型时定义它。

“Python 2 张量流图像将很快被弃用,并且可能不支持即将推出的更新版本的张量流图像。 请设置参数 "py_version='py3'" 以使用 Python 3 张量流图像

SageMaker 目前尚不支持 TensorFlow 2.0(py2 或 py3 均不支持)。但它很快就会与 SageMaker 一起使用。

关于Python版本。目前支持的TensorFlow版本支持py2,但是2020年1月1日之后,以后所有的框架版本将不再支持py2。