GPU部署AI绘画实践-stable-diffusion-webui

登录云GPU机器后，会默认进行驱动的自动安装，本地部署请自行官网下载：

ps aux | grep -i install

自动安装顺序为： （1）GPU驱动； （2）CUDA； （3）cuDNN

当GPU驱动、CUDA、cuDNN都安装完成后，通过以下命令创建python3虚拟环境，完成前置配置。

sudo apt install wget git python3 python3-venv

获取代码

• 将webui clone到服务器本地：

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

• 成功后，进入stable-diffusion-webui文件夹：

cd stable-diffusion-webui

修改配置

在launch.py的代码中，由于腾讯源没有torch==1.13.1+cu117，导致需要从--extra-index-url的 https://download.pytorch.org/whl/cu117 中进行下载，速度非常慢。故此处对 launch.py 代码进行修改：

原代码：torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117")

修改后代码：torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==1.13.1 torchvision==0.14.1")

可通过如下代码实现上述内容

export TORCH_COMMAND="pip install torch==1.13.1 torchvision==0.14.1"

下载模型

• 进入stable-diffusion-webui/models/Stable-diffusion文件夹，
• 通过命令行下载模型到本地。该步骤下载速度较快，
• 大约需要3-5min即可完成。
• 可访问 https://huggingface.co/CompVis/stable-diffusion-v-1-4-original 或 https://civitai.com 进行下载。

cd stable-diffusion-webui/models/Stable-diffusion

wget -O pastelMixStylizedAnime_pastelMixPrunedFP16.safetensors https://civitai.com/api/download/models/6297

下载其他模型模版：wget -O 模型名称 模型下载链接

完成安装

返回进入stable-diffusion-webui文件夹，使用命令 ./webui.sh 启动。

通过自动化脚本完成安装。该步骤会自动安装webui所需的各依赖项，如gfpgan，k_diffusion等。

实测 完成全部安装大概需要10-20min。

./webui.sh

使用外网链接，在./webui.sh后增加 “–share”即可

./webui.sh --share

访问链接

等待10分钟左右，全部安装完成后就可以看到链接了

提示: 由于服务器和github连接不是很稳定，中途可能出现网络错误。若出现网络问题，重新输入“./webui.sh”即可。或可通过报错提示信息，手动执行相应步骤安装后，重新输入“./webui.sh”继续剩余步骤的自动安装。

将上个步骤最后生成的链接粘贴至浏览器即可完成实践内容。