如何在运行时添加外部Python模块以加载PyTorch模型？

AI 小北

2026-8-5

解决方案

当然可以从外部文件夹加载依赖模块，以下是几种可行的方法：

正确配置sys.path路径
你之前的问题大概率是添加的路径不对。假设你的外部模块存放在/path/to/external_modules目录下，目录结构如下：
```
/path/to/external_modules/
    module_a/
        __init__.py
        相关代码文件
    module_b/
        __init__.py
        相关代码文件
```
你需要把外部模块的父目录添加到sys.path，而不是模块本身的路径。代码示例：
```
import sys
# 替换为你的外部模块父目录的绝对路径
sys.path.insert(0, "/path/to/external_modules")

# 之后即可正常导入模块
import module_a
import module_b
```
这样PyTorch加载模型时就能找到依赖的模块了。

动态导入外部模块（无需修改sys.path）
如果你不想全局修改sys.path，可以用importlib.util动态加载指定路径的模块，示例如下：

import importlib.util
import os

def load_module(module_name, parent_dir):
    # 如果是单文件模块
    module_path = os.path.join(parent_dir, f"{module_name}.py")
    # 如果是包，指向__init__.py
    # module_path = os.path.join(parent_dir, module_name, "__init__.py")
    
    spec = importlib.util.spec_from_file_location(module_name, module_path)
    module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(module)
    return module

# 加载外部的module_a
module_a = load_module("module_a", "/path/to/external_modules")

通过环境变量PYTHONPATH配置
在运行脚本前，将外部模块的父目录添加到PYTHONPATH环境变量中，这样Python会自动识别该路径下的模块：
- Linux/macOS终端：
```
export PYTHONPATH="/path/to/external_modules:$PYTHONPATH"
python your_script.py
```
- Windows命令提示符：
```
set PYTHONPATH=C:\path\to\external_modules;%PYTHONPATH%
python your_script.py
```