前言

随着人工智能的不断发展，尤其是chatGPT的成熟，AI绘画也获得了热捧，甚至AI画的太空歌剧院（Midjourney所作）还获得了奖项。

由于线上的绘画需要费用，不够尽兴和个性化，本文就介绍了流行的开源AI绘画程序stable diffusion的本地安装部署。

一、AI绘画是什么？

AI绘画主要包含两个部分，一个是对图像的分析与判断，即“学习”，一个是对图像的处理和还原，即“输出”。人工智能通过对数以万计的图像及绘画作品进行不断学习，现如今，已经达到只需输入清晰易懂的文字标签，即可在很短的时间内得到一张同样效果不错的画面（文生图）。对使用者没有任何绘画功底需求，甚至还能根据要求对画面风格进行精细的改变调整。

AI绘画，顾名思义就是利用人工智能进行绘画，是人工智能生成内容（AIGC）的典型应用场景之一。其主要原理是收集大量已有作品，通过算法对其内容和风格特征进行解析，最后再生成新的作品，所以算法是AI绘画的核心。目前比较流行的算法（AI绘画模型）有：Stable Diffusion，Disco Diffusion，Midjourney，DALL-E 2，NovelAI，前两者的模型是开源的，而后三者未开源。

二、Stable Diffusion简介

Stability.ai公司于2022年8月份发布并开源了Stable Diffusion AI 绘画模型，Stable Diffusion采用的底层算法与普通的扩散算法不同，它在其外面还套了一个VAE模型，因此达到了更快的画图速度和更高的画图质量。Stable Diffusion 在github开源，代码可以直接clone了。

二、部署步骤

1.克隆代码

Git clone https://github.com/AUTOMATIC/stable-diffusion.git

`2.部署环境`

2.1 硬件环境

必要条件： nvidia 的高档显卡，显存8G以上吧。

2.2 软件环境

必要条件：cuda，python，pytorch。

3. 安装

1）安装nvidia显卡驱动。

运行如下指令查看显卡驱动。

nvidia-smi

、结果如下图。

红线所示即所需要的cuda版本。

2) 根据显卡驱动安装对应的cuda torch。

使用pip安装可以去pytorch官网设定命令行，版本18也支持12.0。

3）下载预训练模型。

4）运行webui.bat

5）插件安装部署。如汉化插件等。

汉化插件可以从GitHub - dtlnor/stable-diffusion-webui-localization-zh_CN: Simplified Chinese translation extension for AUTOMATIC1111's stable diffusion webuiSimplified Chinese translation extension for AUTOMATIC1111's stable diffusion webui - GitHub - dtlnor/stable-diffusion-webui-localization-zh_CN: Simplified Chinese translation extension for AUTOMATIC1111's stable diffusion webuihttps://github.com/dtlnor/stable-diffusion-webui-localization-zh_CN克隆或下载，下载后文件放入extensions/stable-diffusion-webui-localization-zh_CN。然后重启UI就可以设置为中文了。

4 一些错误提示的排除

1）GPU不能找到。

更新驱动或者匹配cuda版本，cuda118可以支持12.1的cuda。

2）gfpgan、clip不能安装

使用pip 安装gfpgan、clip。

3）open_clip不能安装

使用其他源安装，如

pip install git+https://gitee.com/ufhy/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b --prefer-binary 作者：貓妖 https://www.bilibili.com/read/cv21816965/ 出处：bilibili