Stable Diffusion WebUI AMDGPU 项目在 RX580 显卡上的 ZLUDA 兼容性问题解决方案

问题背景

在 Windows 10 系统下使用 AMD RX580 显卡运行 Stable Diffusion WebUI AMDGPU 项目时，用户遇到了模型加载失败的问题。该项目旨在为 AMD 显卡用户提供更好的 Stable Diffusion 支持，而 ZLUDA 是一个允许 CUDA 代码在 AMD GPU 上运行的工具。

问题现象

当用户尝试在 RX580 显卡上使用 ZLUDA 运行 Stable Diffusion WebUI AMDGPU 时，虽然界面能够正常启动，但在加载模型检查点（checkpoint）时会出现以下关键错误：

1. 设备初始化阶段报告"ZLUDA device failed to pass basic operation test"
2. 模型加载过程中出现"RuntimeError: Expected all tensors to be on the same device"错误
3. 最终导致"Stable diffusion model failed to load"

根本原因分析

经过技术分析，这个问题主要由以下几个因素导致：

1. Polaris 架构显卡的内存管理问题：RX580 采用的 Polaris 架构在驱动程序处理系统内存时存在已知问题，容易触发内存不足错误。

2. Torch 版本兼容性问题：最新版本的 PyTorch 与 RX580 显卡的 ZLUDA 实现存在兼容性问题。

3. 设备一致性检查失败：模型加载过程中，系统检测到张量被分散在不同设备（CUDA 和 CPU）上，违反了 PyTorch 的运行要求。

解决方案

方案一：使用专业版驱动程序

1. 完全卸载现有的 AMD 显卡驱动（推荐使用 Display Driver Uninstaller）
2. 安装最新版的 AMD PRO Edition 专业驱动程序
3. 重新安装 HIP SDK

方案二：降级 PyTorch 版本（推荐）

对于大多数 RX580 用户，降级 PyTorch 和相关库版本是最有效的解决方案：

1. 在虚拟环境中执行以下命令降级相关包：

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

2. 确保安装的版本完全匹配
3. 清理并重建 Python 虚拟环境

验证方法

用户可以通过以下方式验证 ZLUDA 是否正常工作：

1. 使用 Blender 3D 等支持 CUDA 的应用程序进行渲染测试
2. 在 Stable Diffusion WebUI 中尝试生成简单图像
3. 检查系统日志中是否还有设备不匹配的错误

技术建议

1. 对于 Polaris 架构显卡用户，建议始终使用经过验证的 Torch 版本组合
2. 在出现内存相关错误时，可以尝试添加 --medvram 或 --lowvram 参数启动 WebUI
3. 定期检查 AMD 专业驱动程序的更新，这些驱动通常对专业应用有更好的支持

总结

AMD RX580 显卡用户在使用 Stable Diffusion WebUI AMDGPU 项目时，通过合理选择驱动版本和 PyTorch 版本组合，完全可以实现稳定的 ZLUDA 支持。特别是降级 PyTorch 到 2.0.1 版本配合 torchvision 0.15.2 的方案，在实际使用中表现最为稳定。这一解决方案不仅适用于 Stable Diffusion WebUI AMDGPU 项目，也可应用于其他需要 ZLUDA 支持的 AI 绘画工具。