将数据从 S3 存储桶下载到 SageMaker 推理容器

Question

我将模型部署到 SageMaker 端点以进行推理。我的输入数据非常大，我想将其 S3 URI 发送到端点，以便我可以将其下载到已部署的 Docker 容器中。不幸的是，当我尝试使用 SageMaker SDK 下载数据时，出现此错误：

Read-only file system: '/opt/ml/model/'

如果有人能帮我解决这个问题，我将不胜感激。

Answer 1

SageMaker Inference 实例有一个名为 /temp/ 的目录，该目录是可写的，可用于非持久性存储。我使用 SageMaker SDK 中的 S3Downloader 实用程序将数据下载到此目录。例如：

from sagemaker.s3 import S3Downloader
from sagemaker.session import Session

sagemaker_session = Session()

def download_data_from_s3(s3_uri):
    S3Downloader.download(s3_uri=s3_uri,
                          local_path='/tmp/',
                          sagemaker_session=sagemaker_session)