Windows10 x64环境运行YOLO推理脚本导入torch遇WinError126错误

问题详情 #

在Windows 10 x64系统的Python虚拟环境中，使用ultralytics运行YOLO推理脚本时，导入torch触发如下错误：

Traceback (most recent call last):
  File "C:\Users\apoor\Desktop\football_analysis\yolo_inference.py", line 1, in <module>
    import torch
  File "C:\Users\apoor\Desktop\football_analysis\cv_env\Lib\site-packages\torch\__init__.py", line 148, in <module>
    raise err
OSError: [WinError 126] The specified module could not be found. Error loading "C:\Users\apoor\Desktop\football_analysis\cv_env\Lib\site-packages\torch\lib\fbgemm.dll" or one of its dependencies.

已尝试的操作：

• 通过pip重新安装ultralytics和torch
• 安装x86和x64版本的Microsoft Visual C++可再发行组件
• 确认虚拟环境已正确激活
• 在VS Code中选择了正确的Python解释器

额外解决建议 #

1. 针对性重新安装torch及依赖 #

卸载当前torch后，使用匹配系统的指定版本安装包重新部署，避免自动安装的版本兼容性问题：

• 执行卸载命令：pip uninstall torch ultralytics -y
• 根据你的Python版本和是否使用CUDA，使用对应版本的安装命令（示例为适配CUDA 11.8的版本，需替换为匹配你环境的版本）：

  pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 --index-url https://download.pytorch.org/whl/cu118
  

• 重新安装ultralytics：pip install ultralytics

2. 定位具体缺失的依赖库 #

使用Dependency Walker工具分析fbgemm.dll，明确缺失的具体组件：

• 加载该dll文件后，查看工具中红色标记的缺失项
• 根据缺失项的名称，从微软官方渠道下载安装对应的系统组件

3. 检查虚拟环境路径 #

确保虚拟环境所在路径无中文、空格或特殊符号，这类字符可能导致动态链接库加载失败

4. 更新显卡驱动（CUDA版本适用） #

如果安装的是CUDA版本的torch，更新显卡驱动至最新版本，旧驱动可能引发CUDA相关依赖加载异常

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