如何使用Python SDK调用Amazon SageMaker端点

Question

我正在尝试使用这个非常简单的命令：

import boto3
client = boto3.client('sagemaker-runtime')

列在documentation

但是我收到了这个错误：

UnknownServiceError: Unknown service: 'sagemaker-runtime'. Valid service names are: acm, etc..

我的目标是能够调用我在Amazon SageMaker中创建的端点。

我是在Sagemaker的Jupyter笔记本上做的，所以我觉得这应该没问题。如何让它在Sagemaker环境之外运行？

Answer 1

Amazon SageMaker是一项非常新的服务（2017年12月）。

您需要更新您的boto库才能使用它：

sudo pip install boto --upgrade
sudo pip install boto3 --upgrade
sudo pip install awscli --upgrade

Answer 2

文档不正确。这就是您使用SageMaker Python SDK获取客户端的方法。

import boto3
client = boto3.client('runtime.sagemaker')

我成功完成了这个。并且，正如John所说，请务必更新您的boto3和awscli版本。