SD.Next项目中使用ZLUDA在AMD GFX906显卡上的问题分析与解决方案

背景介绍

SD.Next是一个基于Stable Diffusion的AI图像生成项目，它支持多种硬件加速方案。对于AMD显卡用户而言，ZLUDA是一个能够将CUDA代码转换为在AMD显卡上运行的兼容层工具，理论上可以让AMD显卡获得类似NVIDIA CUDA的运行体验。

问题现象

在Windows 10系统上使用AMD Radeon Pro VII显卡(GFX906架构)运行SD.Next项目时，当启用ZLUDA支持后，生成的图像会出现混乱失真的情况。值得注意的是，系统日志中并未报告任何错误信息，这使得问题排查更加困难。

环境配置

• 操作系统：Windows 10 22H2
• 显卡型号：AMD Radeon Pro VII (GFX906架构)
• 软件版本：
  • SD.Next最新开发版
  • HIP SDK 5.7.1
  • ZLUDA 3.5
• 模型：基于SD 1.5的Anything V5模型

问题分析

通过对比测试发现：

1. 使用DirectML后端时，GFX906显卡能够正常生成图像
2. 使用ZLUDA但强制使用CPU时，也能获得正确结果
3. 只有在ZLUDA+GFX906组合下才会出现图像混乱

这表明问题很可能出在ZLUDA对GFX906架构的支持上，而非SD.Next项目本身。日志显示系统正确识别了GFX906设备，但可能缺少针对该架构优化的内核二进制文件。

解决方案探索

1. 使用预编译的ROCm库

尝试使用社区提供的ROCmLibs库替换官方ROCm库，这个库旨在支持更多GFX架构，包括GFX906。虽然测试中未能解决问题，但这仍是一个值得尝试的方向。

2. 自行编译内核二进制

对于高级用户，可以考虑自行编译针对GFX906架构优化的内核二进制文件。这需要：

• 熟悉ROCm工具链
• 了解AMD GPU架构特性
• 具备一定的编译经验

编译过程涉及：

• 设置正确的编译环境
• 针对特定GFX架构进行优化
• 生成兼容的内核二进制文件

3. 社区协作开发

SD.Next社区正在积极开发支持更多GFX架构的内核二进制文件。用户可以关注社区动态，特别是针对GFX906架构的支持进展。

技术建议

1. 对于GFX906显卡用户，目前建议暂时使用DirectML后端
2. 关注ZLUDA和ROCm对GFX906架构支持的更新
3. 考虑参与社区协作，共同完善对GFX906的支持
4. 在测试新方案时，建议做好环境备份

总结

AMD显卡在AI计算领域的支持仍在不断完善中。GFX906架构在SD.Next项目中通过ZLUDA运行时出现的问题，反映了特定架构支持方面的挑战。随着社区的努力和技术的进步，这些问题有望得到解决。对于遇到类似问题的用户，建议保持耐心，关注官方和社区的更新动态。