AMD显卡如何安装stable-diffusion-webui

文 / @WordPress主题

在日新月异的科技世界中,AMD显卡以其卓越的性能和稳定的兼容性受到了广大用户的青睐,特别是在处理图形密集型任务时表现出的优秀性能让众多设计师对它青睐有加。其中,Stable Diffusion,一个使用AI技术生成图片的软件,就是这样一款能够充分发挥AMD显卡优势的工具。但是,如何在AMD显卡上安装并运行Stable Diffusion呢?今天,我们将带您一步步解读在Windows、Linux,包括Arch Linux系统上,安装并运行Stable Diffusion的webui的详细过程。无论您是初次接触还是经验丰富的用户,都能从本文中找到合适的解决方案,让您的AMD显卡和Stable Diffusion完美结合。

photo-1606963060045-1e3eaa0e6eac

在AMD显卡上安装和运行Stable Diffusion的webui有多种方法,可以根据您的操作系统选择最适合的方式进行。

在Windows系统上安装

目前Windows+AMD的官方支持尚未发布,不过您可以安装由lshqqytiger创建的webui分支版本,它采用Direct-ml。安装步骤如下:

  1. 安装Python 3.10.6(勾选Add to PATH)和git。
  2. 在cmd/终端中粘贴以下代码行:

    git clone https://github.com/lshqqytiger/stable-diffusion-webui-directml && cd stable-diffusion-webui-directml && git submodule init && git submodule update

  3. 双击webui-user.bat文件。

注意,若在安装或运行时出现卡顿现象,请在终端中按Enter键,然后程序应会继续进行。

在Linux系统上安装

  1. 安装Python 3.10.6。
  2. 运行以下命令克隆仓库:

    git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui

  3. 将Stable Diffusion模型检查点(model.ckpt)放入models/Stable-diffusion目录中。
  4. 根据您的显卡型号,可能需要在webui-user.sh中为COMMANDLINE_ARGS=添加参数--precision full --no-half,以避免出现黑方块或崩溃的情况。
  5. 运行webui.sh。

注:部分显卡,如Radeon RX 6000系列和RX 500系列,可以不使用--precision full --no-half选项,这样可以节省大量显存。

在Arch Linux上安装

以下是在Arch Linux以及可能的其他基于Arch的Linux发行版上安装webui的方法(测试于2023年2月22日)。

  1. 首先,使用以下命令安装必要的依赖项并安装pip:

    sudo pacman -S python-pip

  2. 安装具有ROCm后端的pytorch:

    选择并安装以下两个包之一:python-pytorch-rocmpython-pytorch-opt-rocm(仅对AVX2 CPU)。

  3. 安装具有ROCm后端的torchvision:

    从AUR克隆git仓库并在本地编译安装python-torchvision-rocm包。

  4. 设置venv环境:

    手动创建一个带有系统site-packages的venv环境(这将允许访问系统的pytorch和torchvision),安装剩余的Python依赖项。

  5. 创建webui启动脚本。
  6. 运行以下命令以启动webui:

    source venv/bin/activate

    ./webui-py.sh

注:根据您的GPU型号,可能需要添加某些命令行参数和优化项,以使webui正常运行。

上述安装方法均已经在AMD Radeon RX6800、Python 3.10.9、ROCm 5.4.3、PyTorch 1.13.1、Torchvision 0.14.1环境下测试过。

安装过程中可能会出现一些限制,例如GPU型号需要由Arch依赖项支持,同时Arch依赖项(pytorch、torchvision)的更新将由全系统更新(pacman -Syu)和编译来维护,这可能不适用于希望使用固定版本组合的依赖项的情况。如有任何问题,可考虑使用上述其他的安装方法。

添加UTHEME为好友
扫码添加UTHEME微信为好友
· 分享WordPress相关技术文章,主题上新与优惠动态早知道。
· 微信端最大WordPress社群,限时免费入群。