当前位置:AIGC资讯 > AIGC > 正文

Stable-Diffusion AI画画本地搭建详细步骤

ChatGPT出来后,第一次感觉到人工智能真的可能要来了,因此也顺便尝试了下开源AI画画的搭建。网络上写的教程总是不那么面面俱到,因此本文参考了3篇文章才成功把Stable-Diffusion 本地搭建搭建了起来。参考教程在文末。

   本文是本地搭建AI画画,也就是用自己的显卡去进行AI画画,教程1是第一次找到的,讲的是利用云端GPU生成AI画画,显然不是我们想要的。错误的项目地址如下:GitHub - camenduru/stable-diffusion-webui-colab: stable diffusion webui colab

参考教程2,我得到了正确的项目地址,将这个项目下载下来:GitHub - AUTOMATIC1111/stable-diffusion-webui: Stable Diffusion web UI

Conda虚拟环境

        看教程需要conda这种东西,身为Java程序员的我显然一脸懵逼,不过很快就明白了这个命令是需要安装anaconda的,去其他教程2中的清华开源软件镜像站将“Anaconda3-2022.05-Windows-x86_64.exe”下载下来,直接安装即可,接着就是修改conda创建的虚拟环境

搜索anaconda

执行路径切换

conda config --append envs_dirs D:\ai_painting\anaconda_env

创建虚拟环境,环境名是stable_diffusion,使用的python版本是3.10.6,然后激活环境

conda create -n stable_diffusion python=3.10.6

conda activate stable_diffusion

模型文件下载:

      教程2有提供了一些现成的模型,我下了一个wd-v1-3-full.ckpt丢到D:\ai_painting\stable-diffusion-webui-master\models\Stable-diffusion里面

安装Cuda:

我的显卡是3070ti,因此命令行输入nvidia-smi,查看3070ti的cuda版本,发现是11.7

然后去官网下载:CUDA Toolkit Archive | NVIDIA Developer 

找个11.6的吧。CUDA Toolkit 11.6 Update 2 Downloads

运行项目:

      进入项目文件夹,编辑webui-user.bat,修改如下,要填入anaconda创建的虚拟环境的python的路径,以及电脑git软件的位置,以及加上一个命令行参数--no-gradio-queue

运行webui-user.bat,会开始下一些东西到C盘

比如torchvision是pytorch的一个图形库,它服务于PyTorch深度学习框架的,主要用来构建计算机视觉模型。python torch又称PyTorach,是一个以Python优先的深度学习框架,一个开源的Python机器学习库,用于自然语言处理等应用程序

耐心等待其下载完成,中途可能会遇到卡在安装gfpgan,解决方法各种各样我就不说了自己百度去,原因很简单就是因为有墙壁。

测试运行:

 浏览器输入:http://127.0.0.1:7860

输入a beautiful girl,就生成了似乎很丑的girl图片,可能是模型库不好吧

 GPU主要利用的是专用内存

 显卡似乎不太给力,温度达到了90度

 这AI画画生成的图片挺恐怖的

参考教程1:保姆级教程教你快速搭建属于自己的AI绘画系统!(收藏)_大数据之眸的博客-CSDN博客

参考教程2:(二十三)——window10 AI绘画 stable-diffusion-webui安装 - 知乎 (zhihu.com)

参考教程3:从零开始,win10本地部署stable-diffusion-webui AI绘图 - 知乎 (zhihu.com)

其他教程1:Anaconda的下载和使用方法_厦大数据库实验室博客 (xmu.edu.cn)

其他教程2:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

其他教程3:anaconda新建虚拟环境于指定路径_conda create 指定路径_Reza.的博客-CSDN博客

其他教程4:

求救,卡在这里了【ai绘画吧】_百度贴吧 (baidu.com)

本地部署AI作画工具stable-diffusion-webui(Novel AI)报错排查 - 哔哩哔哩 (bilibili.com)

其他家教程5:stable-diffusion-webui 安装问题记录 - 哔哩哔哩 (bilibili.com)

更新时间 2023-11-22