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)