Stable Diffusion WebUI DirectML 在 AMD RX 6650XT 显卡上的配置指南

问题背景

在使用 Stable Diffusion WebUI DirectML 项目时，部分 AMD RX 6650XT 显卡用户遇到了图像生成失败的问题。错误日志显示主要与"LayerNormKernelImpl"和"Half"数据类型相关的运行时错误。

错误分析

从技术角度来看，这些错误通常源于以下原因：

1. 错误的启动参数配置
2. 数据类型精度不兼容问题
3. 显卡驱动与计算框架的适配问题

具体表现为：

• "LayerNormKernelImpl not implemented for 'Half'"错误
• "addmm_impl_cpu_ not implemented for 'Half'"错误
• "log_vml_cpu not implemented for 'Half'"错误

解决方案

正确的启动参数配置

关键解决方法是修改启动参数：

1. 移除 --skip-torch-cuda-test 参数
2. 添加 --use-directml 参数

这个调整确保了：

• 正确启用 DirectML 后端
• 避免不必要的 CUDA 测试
• 确保使用兼容的数据类型

其他可能需要的调整

1. 更新显卡驱动：确保使用最新版的 AMD 显卡驱动
2. 检查 Python 环境：确认使用的是兼容的 Python 版本
3. 验证模型文件：确保下载的模型文件完整无损坏

技术原理

DirectML 是微软开发的 DirectX 12 的机器学习组件，专为 Windows 平台优化。当使用 AMD 显卡时：

1. DirectML 提供了比 CUDA 更优的兼容性
2. 避免了 CUDA 相关的错误检测
3. 提供了更好的半精度浮点运算支持

使用建议

1. 对于 AMD 显卡用户，始终优先使用 --use-directml 参数
2. 避免同时使用 --skip-torch-cuda-test 和 --use-directml 参数
3. 首次运行时耐心等待模型加载完成

总结

通过正确配置启动参数，AMD RX 6650XT 显卡用户可以顺利运行 Stable Diffusion WebUI DirectML 项目。关键在于使用专为 DirectML 优化的参数配置，避免 CUDA 相关的兼容性问题。这一解决方案不仅适用于 RX 6650XT，也适用于大多数 AMD 显卡用户。