在 sagemaker 中从 S3 自定义安装 package

Question

我在 S3 存储桶中有一个自定义 package（未发布的开源）的 whl 文件。

我现在想在我的 sagemaker 实例中导入/安装它。 https://medium.com/@shadidc/installing-custom-python-package-to-sagemaker-notebook-b7b897f4f655此链接是我尝试访问的链接，但它对我不起作用。

有人试过这个吗？

Answer 1

使用 AWS Sagemaker Jupyter 单元

选项1：

快速而肮脏，将 whl 文件上传到与笔记本相同的工作区并安装。

%pip install custom_package_name.whl

现在，您需要重新启动 kernel 然后再导入。 你应该能够完成。

选项 2：

另一种简单的方法是从 S3 文件夹中获取 whl 文件并安装。

s3 = boto3.client("s3")
s3_bucket = "your core s3 bucket location"
file_location = "your file location and file name with extension"

下面我为源和目标使用相同的文件位置

s3_client.download_file(s3_bucket, file_location, file_location)

%pip install ./your_target_folder_name/custom_package.whl

现在，您需要重新启动 kernel 然后再导入。 你应该能够完成。