AMD GPU安装运行stable diffusion

[复制链接]
查看542 | 回复0 | 2023-8-16 15:23:39 | 显示全部楼层 |阅读模式
本文操作环境为Windows10/11+AMD
AI绘画是一种利用人工智能技术进行绘画的方法。它可以通过机器学习算法来学习艺术家的风格,并生成类似于艺术家的作品。最近,AI绘画技术得到了很大的发展,许多公司和研究机构都在进行相关的研究和开发。例如,OpenAI最近推出了一个名为DALL-E的项目,它可以根据用户提供的文本描述生成图像。此外,AI绘画技术还被应用于游戏、电影、广告等领域,以及艺术家们的创作中。而Stable diffusion是一种用于图像生成的算法,它可以生成高质量的图像。
1,安装前说明
   Windows+AMD的支持还没有正式为webui做,
但我们可以安装 lshqqytiger 的使用 Direct-ml 的 webui 分支。
  仓库地址如下:
https://github.com/lshqqytiger/stable-diffusion-webui-directml/issues
2,安装
1)安装Python3.10.6
  1. 如果已经装了可以卸载原版本后下载3.10.6版
复制代码
下载地址如下:
https://www.python.org/downloads/release/python-3106/
根据自己的配置选择下载

tips:安装的时候记得勾选 “添加到环境变量”
2)复制项目到本地
  1. git clone https://github.com/lshqqytiger/stable-diffusion-webui-directml
  2. cd stable-diffusion-webui-directml
  3. git submodule init
  4. git submodule update
复制代码
3)下载并添加模型
下面的链接,进去选择自己喜欢的模型下载好后放入models目录下的Stable-diffusion文件夹内。
https://huggingface.co/models
4)有部分amd的卡会出现不兼容的情况,可以选择在webui-user.sh文件中加入下面的内容
  1. --precision full--no-halfCOMMANDLINE_ARGS=
复制代码
5)双击运行webui.sh
稍作等待后终端显示127.0.0.1:7860 直接在浏览器访问就可进入stable diffusion的网页操作页面。

题外话
本机运行
  1. git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
  2. cd stable-diffusion-webui
  3. python -m venv venv
  4. source venv/bin/activate
  5. python -m pip install --upgrade pip wheel
  6. # It's possible that you don't need "--precision full", dropping "--no-half" however crashes my drivers
  7. TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' python launch.py --precision full --no-half
  8. cd stable-diffusion-webui
  9. # Optional: "git pull" to update the repository
  10. source venv/bin/activate
  11. # It's possible that you don't need "--precision full", dropping "--no-half" however crashes my drivers
  12. TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' python launch.py --precision full --no-half
复制代码
第一次启动 WebUI 可能需要很长时间,可能会看到类似这样的提示:
   MIOpen(HIP): Warning [SQLiteBase] Missing system database file: gfx1030_40.kdb Performance may degrade. Please follow instructions to install: https://github.com/ROCmSoftwarePlatform/MIOpen#installing-miopen-kernels-package
  在Docker中运行。
拉取最新的 Docker 镜像,启动镜像并附加到容器:
  1. rocm/pytorchrocm/pytorchdocker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add=video --ipc=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $HOME/dockerx:/dockerx rocm/pytorch
复制代码
在容器内执行以下命令:
  1. cd /dockerx
  2. git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
  3. cd stable-diffusion-webui
  4. python -m venv venv
  5. source venv/bin/activate
  6. python -m pip install --upgrade pip wheel
  7. # It's possible that you don't need "--precision full", dropping "--no-half" however crashes my drivers
  8. TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' REQS_FILE='requirements.txt' python launch.py --precision full --no-half
复制代码
容器内的文件夹应该可以在主目录中以相同的名称访问。
在 Docker 中更新 Python 版本
如果 Web UI 变得与 Docker 映像中预安装的 Python 3.7 版本不兼容,可以按下面的操作(前提是已经在docker运行成功后)
在容器内执行以下命令:
  1. apt install python3.9-full # Confirm every prompt
  2. update-alternatives --install /usr/local/bin/python python /usr/bin/python3.9 1
  3. echo 'PATH=/usr/local/bin:$PATH' >> ~/.bashrc
复制代码
之后再按照上面在Docker中运行的步骤走一遍,使用下面修改后的启动命令
   rm -rf /dockerx/stable-diffusion-webui/venv
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui

来源:https://blog.csdn.net/qq_43774332/article/details/130213368
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则