在Amazon SageMaker中添加模型依赖报错：找不到module_0文件夹怎么办？

错误原因 #

你在PyTorchModel初始化时的路径配置存在问题：

• source_dir='code_alpha'已经指定了要打包上传的代码目录，该目录下的module_0、module_1会被自动包含在部署包中。
• dependencies参数的作用是指定不在source_dir目录下的额外依赖文件/文件夹。你把已经在source_dir内的子目录加到dependencies里，系统会去当前Notebook的工作目录查找这些文件夹，自然找不到，从而触发报错。

解决方案 #

方案1：移除多余的dependencies参数 #

直接删除dependencies配置项，因为source_dir已经包含了所有需要的代码文件和子目录。修改后的代码如下：

model_alpha = PyTorchModel(
    entry_point='inference_alpha.py',
    source_dir='code_alpha',
    role=role,
    model_data=model_data_alpha,
    framework_version="1.7.1",
    py_version='py3',
    code_location='s3://model',
    name='alpha-model',
)

额外检查点 #

1. 确认Notebook的工作目录是main文件夹（你的目录结构根目录），可以执行以下命令查看当前路径：

!pwd

如果当前路径不是main，需要切换目录：

import os
os.chdir('main')

2. 确保inference_alpha.py中导入模块的方式正确，使用绝对导入而非相对导入，比如：

import module_0
import module_1

内容的提问来源于stack exchange，提问作者Diego Rodea