在 sagemaker Mxnet 1.4.1 映像中安装用于推理的外部库的最佳方法是什么?

Question

我在官方文档中读到require.txt是用来为sagemaker安装库的。但是，它也说：

对于 MXNet 1.4.1，推理不支持要求.txt

我正在使用 MxnetModel 类创建一个模型，指定自定义脚本作为入口点。

安装它的测试方法是什么？ 我应该将 Pip 安装"到由 sagemaker 加载的推理脚本中吗？

谢谢

Answer 1

我建议 pip 将您的依赖项安装到特定文件夹中，然后将它们与您的脚本一起打包在源目录中。您还需要在"依赖项"参数中引用它：

• pip install my_dependency_name -t my_source_directory_path
• 模型([...]， source_dir=my_source_directory_path， 依赖关系=[my_dependency_name] (

下面是一个示例：https://github.com/aws/sagemaker-python-sdk/blob/bf48fb1219bd8ea22e78a913bfa091e544c57cc3/tests/integ/test_rl.py#L90