最近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
版权归原作者 formaever 所有， 如有侵权，请联系我们删除。