Stable-Diffusion-WebUI-AMDGPU项目中的ONNX Runtime DLL加载问题解决方案

问题背景

在使用Stable-Diffusion-WebUI-AMDGPU项目时，部分用户遇到了一个与ONNX Runtime相关的DLL加载失败问题。具体表现为在启动WebUI时出现错误信息："ImportError: DLL load failed while importing onnxruntime_pybind11_state"，提示动态链接库(DLL)初始化程序失败。

问题分析

该问题主要源于ONNX Runtime DirectML 1.18.1版本的一个已知兼容性问题。当项目尝试加载onnxruntime_pybind11_state模块时，系统无法正确初始化相关的动态链接库，导致程序启动失败。

解决方案

经过技术验证，可以通过以下步骤解决此问题：

1. 清理虚拟环境：首先删除项目目录下的venv文件夹，确保干净的安装环境。

2. 重新初始化项目：

   • 进入项目目录
   • 执行webui.bat --use-directml命令（或在webui-user.bat中设置--use-directml参数）

3. 安装特定版本ONNX Runtime：

   • 激活虚拟环境：.\venv\scripts\activate
   • 安装指定版本：pip install onnxruntime-directml==1.18.0
   • 可选步骤：pip install onnxruntime==1.18.0（确保完全兼容）

4. 重新启动项目：使用webui.bat --use-directml或配置好的webui-user.bat启动

技术原理

此问题的根本原因是ONNX Runtime 1.18.1版本在DirectML支持方面存在兼容性问题。通过降级到1.18.0版本，可以规避这个DLL加载失败的问题。ONNX Runtime作为深度学习推理引擎，其稳定性对Stable Diffusion的运行至关重要。

预防措施

为避免类似问题，建议：

1. 在升级ONNX Runtime前，先备份当前工作环境
2. 关注ONNX Runtime官方发布的已知问题列表
3. 对于生产环境，建议固定关键依赖的版本号

总结

通过降级ONNX Runtime到1.18.0版本，可以有效解决Stable-Diffusion-WebUI-AMDGPU项目中的DLL加载失败问题。这体现了深度学习项目中依赖管理的重要性，特别是在使用特定硬件加速方案时，版本兼容性需要格外注意。