Stable Diffusion WebUI DirectML 项目中的 AMD GPU 兼容性问题解析

在 Stable Diffusion WebUI DirectML 项目中，用户在使用 AMD Radeon RX 7900 XT 显卡时遇到了 RocBlas 错误，错误信息显示系统错误地识别了 GPU 架构为 gfx1036 而非正确的 gfx1100。本文将深入分析这一问题的原因及解决方案。

问题现象分析

当用户尝试在 Windows 系统上使用 AMD Radeon RX 7900 XT 显卡运行 Stable Diffusion WebUI 时，系统报出 RocBlas 错误，提示无法读取特定路径下的 TensileLibrary.dat 文件。错误信息中显示系统识别到的 GPU 架构为 gfx1036，而实际上 RX 7900 XT 应该支持的是 gfx1100 架构。

根本原因

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

1. 多 GPU 设备冲突：系统同时检测到了集成显卡和独立显卡，导致 ZLUDA 无法正确识别目标显卡。

2. 架构识别错误：系统错误地将 RX 7900 XT 识别为 gfx1036 架构而非其实际的 gfx1100 架构。

解决方案

针对这一问题，开发者提供了明确的解决方案：

1. 使用 HIP_VISIBLE_DEVICES 环境变量：通过设置该环境变量可以强制指定 ZLUDA 使用的 GPU 设备。例如，设置 HIP_VISIBLE_DEVICES=1 将使 ZLUDA 仅识别和使用 GPU 1（通常是独立显卡）。

2. 单显卡运行原则：ZLUDA 在同一时间只能支持一个显卡设备运行，因此需要确保系统不会同时尝试使用多个 GPU。

技术背景

ZLUDA 是一个允许 CUDA 代码在 AMD GPU 上运行的技术实现。在 Windows 环境下，当系统存在多个 GPU（如集成显卡+独立显卡）时，可能会出现设备识别冲突。HIP_VISIBLE_DEVICES 环境变量是 AMD ROCm 平台提供的标准方法，用于控制可见的 GPU 设备。

实施建议

对于遇到类似问题的用户，建议采取以下步骤：

1. 确定系统中各 GPU 设备的编号（可通过设备管理器或相关诊断工具查看）

2. 在启动 WebUI 前设置正确的 HIP_VISIBLE_DEVICES 环境变量

3. 确保系统只使用目标独立显卡运行 Stable Diffusion

4. 验证 GPU 架构识别是否正确

总结

AMD 显卡在 Windows 环境下运行 Stable Diffusion 时可能会遇到架构识别和多 GPU 冲突问题。通过合理使用 HIP_VISIBLE_DEVICES 环境变量可以有效地解决这些问题，确保 ZLUDA 技术能够正确识别和使用目标显卡。这一解决方案不仅适用于 RX 7900 XT，对于其他 AMD 显卡在多 GPU 环境下的使用也具有参考价值。