(转!)A卡玩转Stable Diffusion

目录

  其实，用A卡也能跑Stable Diffusion，具体操作步骤如下：
  总结：

转：https://www.fujieace.com/ai/amd-stable-diffusion.html

 

今天，一位网友通过查看“设备管理器”，发现他的显卡是“AMD Radeon RX 580 2048SP”。

 

因此，Stable Diffusion用显卡却怎么也跑不起来？一生成图片就报错误“rocBLAS error: Could not initialize Tensile library”，结果只能用CPU来跑Stable Diffusion。

 

其实，用A卡也能跑Stable Diffusion，具体操作步骤如下：

此方法不需要你懂太多的计算机知识，而且仅仅只是在整合包的基础上做了小小的参数设置而已。打破传统Stable Diffusion只能用NVIDIA显卡，不能用AMD显卡的认识。

原理

用的项目源码是：https://github.com/lshqqytiger/stable-diffusion-webui-directml

不同的是，此源码使用：

• --use-directml: DirectML作为torch后端。
• --use-zluda：使用ZLUDA作为torch后端。
• 支持ONNX运行时。
• 支持Olive模块优化。

 

 

1、确保你已经搭建好Stable Diffusion WebUI。

具体可先参考：Stable Diffusion 下载+安装+使用教程

 

2、进入SD目录，双击“A启动器.exe”启动绘世。

例如：G:\AI 工具 Stable Diffusion\sd-webui-aki-v4

 

3、点击“设置”按钮图标，进入“一般设置”界面，找到“配置模式”，选择“专家”选项。

 

 

4、点击“版本管理 ”按钮图标，进入“内核”界面，

找到“当前分支 - 切换分支”，

选择“Ishqqytiger/stable-diffusion-webui-directml - 主线”选项。

关于为什么选择这个？那是因为我测试其它的，我的AMD Radeon RX 580 2048SP显卡和它不兼容，其它的都用不了，只有这一个分支可以和我的A卡兼容。

 

 

5、再点击“切换分支”按钮。

 

6、会提示：切换分支可能导致文件丢失，请提前备份重要文件。您确定要继续吗？直接点击“确定”按钮。

 

 

7、切换完分支后重新再次进入，选择“版本管理”图标按钮，选择”开发版“，切换到”2023年10月28号”那个版本。

这是是因为我试了好几个版本，我能用的只有2023年10月28号的那个版本，只有此版本才我的AMD Radeon RX 580 2048SP显卡兼容。

 

 

8、切换好版本后，再次重新进入，选择“高级选项”图标按钮，“生成引擎”这里这里就会多DML开头的显卡了，再选择“DML GPU 0：AMD Radeon RX 580 2048SP (8G)”这个选项。

 

 

9、后面就很简单了，运行后，点击“一键启动”按钮会让你装一个高版本的PyTorch，只需要点击“前往高级选项安装 PyTorch”即可。

环境完整性检测

需要行动 - 检测到无效配置

您当前安装的 PyTorch 及其附属库并不支持使用您在“生成引擎”中所选择的设备，继续运行将导致启动失败。

请前往“高级选项 - 环境维护”面板安装对应 PyTorch 版本。

 

 

10、选择版本“Torch 1.13.1 （DirectML）”选项，再点击“安装”按钮，等安装成功即可！

其它版本也可以试试，例如：Torch 2.0.0 （DirectML）；

 

 

11、再次重新一键启动就可以直接用A卡跑Stable Diffu

总结：

• 简单来说只有Stable Diffusion开发版才能用A卡，Stable Diffusion稳定版是用不了的。
• 以下是一位网友亲自对我说：我试了一下此方法，现在可以用A卡跑了，速度比CPU快多了。

作者：紫陌花间客
欢迎任何形式的转载，但请务必注明出处。
限于本人水平，如果文章和代码有表述不当之处，还请不吝赐教。