Ubuntu20.04本地部署Stable-Diffusion-webui AI绘画环境

最近AI绘画很火，所以想在本地部署个环境体验一下。原本打算在Windows下安装，网上有很多教程，也有很多大佬做了整合包，可是我的Windows系统下的显卡驱动版本太低，也没法升级，只能装CUDA9，导致安装torch会出现问题。所以考虑在Ubuntu下安装，参考了几篇文章，所幸也安装成功了。这里记录一下安装过程，以供参考。

我的机器配置：

 1.安装cuda

支持安装的cuda版本可以用nvidia-smi命令查看

我的显卡驱动支持的cuda版本最高是11.4，可以在这里下载对应的cuda版本：CUDA Toolkit Archive | NVIDIA Developer。我由于之前安装别的环境，已经装了cuda10.2版本，所以我就直接用这个版本了。可以用nvcc -V查看安装的cuda版本。 

2.搭建python虚拟环境

使用miniconda，我之前已经安装了。然后使用conda创建虚拟环境sdwebui，使用的python版本为3.10.9（stable-diffusion-webui用的python版本为3.10.6）。

conda create -n sdwebui python==3.10.9

激活环境

conda activate sdwebui

升级pip并更改默认库包下载地址为清华镜像

python -m pip install --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

安装torch

在pytorch官网Previous PyTorch Versions | PyTorch

查看对应的cuda支持的torch版本，并且直接拷贝安装命令。

这里是我的cuda版本对应的安装命令。 

3.下载并安装stable-diffusion-webui

切换到当前用户目录下，使用git命令下载stable-diffusion-webui源码

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

由于国内github不能稳定连接，所以可以从国内的gitee（码云）下载

git clone https://gitee.com/yiouyou/automatic1111-stable-diffusion-webui.git

打开v2-1_768-ema-pruned.ckpt · stabilityai/stable-diffusion-2-1 at main，下载训练模型。下载完成后，把v2-1_768-ema-pruned.ckpt这个训练模型放入stable-diffusion-webui的models/Stable-diffusion目录下，这个目录专门存放用于生成AI绘图的绘图元素的基础模型库。后续如果在其他网站比如civitai之类的地方下载的ckpt或者safetensors文件也是放在这个文件夹里面。

 

然后切换到webui目录下，安装依赖

cd stable-diffusion-webui/
pip install -r requirements_versions.txt
pip install -r requirements.txt

安装完成后，启动stable-diffusion-webui

python launch.py

这里要安装很多依赖，并且要git clone相关包和模型文件，需要科学上网，否则会失败。

最终启动成功后如下图所示，会出来一个url

 在浏览器打开这个url，如图所示，表示环境部署成功

 4.使用stable-diffusion-webui

在左上角的文本框prompt输入提示词，点击Generate，等待数秒钟（我的电脑大概等待15s左右），就会生成一张图。

注意：如果在prompt输入后点击generate无法生成图片，看见黑色小框显示“float 32“之类的关键字，则在webui页面的Settings->StableDiffusion最下面，勾选float32的选项框。接着点击上方的Apply settings应用，然后按F5刷新页面后，即可正常使用。

目前只是初步体验了一下，下一步会继续学习，继续解锁新玩法。

参考：

Ubuntu安装stable-diffusion-webui详细教程 - 知乎 

从零开始，手把手教你本地部署Stable Diffusion Webui AI绘画(Win系最新版) - 知乎

其他：

AI绘画咒语tag在线生成器：自己去找对应的tag生成prompt

https://civitai.com：这个里面有很多好看的模型