当前位置:首页|资讯|Stable Diffusion

Archlinux使用Stable-diffusion+ROCm5.6

作者:市ヶ谷有咲official发布时间:2023-09-12

参考:

【A卡ROCm运行Stable Diffusion AI画图-哔哩哔哩】 https://b23.tv/UovFl9a

标注为红色的内容需要可以访问Github

1.安装Archlinux发行版

比如常见的Manjaro,EndeavourOS

可以使用ventory简化安装流程。

在安装过程中如果出现无限加载的情况,可以尝试切断互联网后重试。

2.准备工作
2.1 安装python

sudo pacman -S python python-pip

2.2 准备webui

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

超时与失败是正常的,因为某些原因,可能会需要重复几次

2.3 准备虚拟环境

进入sd目录

python -m venv venv

source venv/bin/activate

pip3 install --upgrade pip wheel

2.4 安装pytorch与torchvision

https://pytorch.org/get-started/locally/

对于比较新的显卡,可以

pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.6

比较旧的显卡可以尝试安装rocm5.5或rocm5.4.2

2.5 尝试运行

打开一个新的终端,进入sd目录

sh webui.sh

稍作等待,此时会下载一些必要的文件,当出现端口号时,说明安装已经成功

3. 更新

3.1 更新webui

进入sd目录,输入

git pull

3.2 更新pytorch与torchvision

进入虚拟环境(参考2.3),

pip3 uninstall pytorch

pip3 uninstall torchvision

之后参考2.4,安装新的torch与torchvision即可。



Copyright © 2024 aigcdaily.cn  北京智识时代科技有限公司  版权所有  京ICP备2023006237号-1