在 AMD 上安装
以下说明仅适用于 Linux!可以在此处(未经测试)找到适用于 Windows 用户的替代指南。
原生运行
执行以下操作:
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
cd stable-diffusion-webui
python -m venv venv
source venv/bin/activate
python -m pip install --upgrade pip wheel
# It's possible that you don't need "--precision full", dropping "--no-half" however crashes my drivers
TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' python launch.py --precision full --no-half
在接下来的运行中,您只需要执行:
cd stable-diffusion-webui
# Optional: "git pull" to update the repository
source venv/bin/activate
# It's possible that you don't need "--precision full", dropping "--no-half" however crashes my drivers
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):警告 [SQLiteBase] 缺少系统数据库文件:gfx1030_40.kdb 性能可能会降低。请关注 安装说明:https://github.com/ROCmSoftwarePlatform/MIOpen#installing-miopen-kernels-package
下一代应该以正常的表现工作。你可以点击消息中的链接,如果你碰巧 要使用相同的操作系统,请按照此处的步骤解决此问题。如果没有明确的编译方式或 为您的操作系统安装 MIOpen 内核,请考虑遵循下面的“在 Docker 中运行”指南。
在 Docker 中运行
拉取最新的 rocm/pytorch
Docker 镜像,启动镜像并附加到容器(取自 rocm/pytorch
文档):docker 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
在容器内执行以下命令:
cd /dockerx
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
cd stable-diffusion-webui
python -m venv venv
source venv/bin/activate
python -m pip install --upgrade pip wheel
# It's possible that you don't need "--precision full", dropping "--no-half" however crashes my drivers
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 container ls --all
,选择与
rocm/pytorch
图像,重新启动它:docker container restart <container-id>
然后附加到它:`docker exec -it