Skip to main content

stable-diffusion-webui

核心流程

  • 搭建PYTHON 环境

  • 搭建CUDA 运行环境

    1. 下载PyTorch最新支持的CUDA驱动:地址

    2. 根据驱动的 NVCUDA64.dll 版本,下载对应的CUDA Toolkit:地址

    3. 下载模型:地址

一、下载最新PyTorch支持的驱动

访问PyTorch官方,查看最新支持的CDUA版本

image-20230409125421901

查看当前驱动的cuda版本于CUDA-api是否匹配

nvidia-smi # 右上角版本于下方版本是否一致

nvcc -V #

进入nvida 驱动程序搜索页:

1080ti下载地址cuda
522.30https://www.nvidia.cn/download/driverResults.aspx/193888/cn/11.8
516.94https://www.nvidia.cn/download/driverResults.aspx/193019/cn/11.7

image-20230409123850201

// 复制下面代码到控制台打开
var numresults = 100 // 每次显示多少个结果
SystemScanner.prototype.DriverSearch = function(psid, pfid, osID, langCode, whql, beta, dltype, numresults ) {numresults=numresults;this.scannerStatusUpdate(GFE_SERVER_CONNECTING);theScanner.scannedDevice.downloadInfo=new Object();var parameters='psid='+psid;parameters+='&pfid='+pfid;parameters+='&osID='+osID;parameters+='&languageCode='+langCode;parameters+='&beta='+beta;parameters+='&isWHQL='+whql;parameters+="&dltype="+dltype;parameters+="&sort1=0";parameters+="&numberOfResults="+numresults;var requestUrl=this.driverManualLookupUrl+parameters;this.driversLogUIEvent("warn","SUID:"+this.tracker.scanID+" BEGIN DriverSearch requestUrl:"+requestUrl);this.debugTrace(requestUrl);jQuery.ajax({url:requestUrl,async:false,type:'get',success:function(response){try{theScanner.debugTrace("The Driver Lookup Service Returned:\n\n("+response+")");if(response.length>0){theScanner.resetResults();var driverLookupJsonObj='('+response+')';theScanner.resultsList=new Object();theScanner.resultsList=eval(driverLookupJsonObj)}if(theScanner.resultsList.Success==0){theScanner.scannerStatus="No driver available"}else{theScanner.scannerStatus="Results Ready"}}catch(e){this.driversLogUIEvent("error"," FAIL catch DriverSearch");theScanner.resetResults();theScanner.scannerStatus="No driver available"}},error:function(response){theScanner.resetResults();theScanner.scannerStatus="AJAX Call failed"}});this.driversLogUIEvent("warn","SUID:"+this.tracker.scanID+" END DriverSearch requestUrl:"+requestUrl);}

下载类型选择Studio 驱动,可以让搜索结果涵盖更多,进入每个版本查看相关文档确定对应的cuda版本

image-20230409124025007

image-20230409124108608

这里可以看到,522.30版本包含的CUDA版本为11.8,下载安装即可。

二、CUDA Toolkit

获取 NVCUDA64.dll 版本

nvidia-smi # 右上角版本于下方版本是否一致

image-20230409114356190

下载 CUDA Toolkit

image-20230409115023601

下载对应的CDUA即可

下载地址:https://developer.nvidia.com/cuda-toolkit-archive

image-20230409113923716

验证安装:

image-20230409125918094

# cmd中输入
nvcc -V

# 查看版本
nvidia-smi.exe

如果返回CUDA工具的版本即为安装成功。

三、模型下载

模型汇总:https://rentry.org/sdmodels

常见二次元模型为NovelAI,Waifu Diffusion,TrinArt。不同的模型会有自己的额外关键词。

Stable-Diffusion官方模型:https://huggingface.co/CompVis/stable-diffusion-v-1-4-original (需要注册并登录hugging face。)

Extra分项:

可使用GFPGAN,ESRGAN等模型用于其他作用。例如GFPGAN可修复脸部,ESRGAN用于放大图片。

GFPGAN v1.4:https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth

下载后放入stable-diffusion-webui/models/gfpgan目录下。

RealESRGAN X4PLUS ANIME 6B: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth

下载后放入stable-diffusion-webui/models/realesrgan目录下。需要在settings里设置让其显示才可以用。

之后再EXTRA选项卡中可以勾选对应模型进行额外操作。

更新webui库:

# 更新依赖库。 
# 拉取新代码 pip install -r requirements.txt
git pull

使用poetry安装

更换变量

set TORCH_COMMAND pip install {whl_path}

set TORCH_COMMAND "pip install D:/CPS/ai/sd/setup/whl/torch-1.13.1+cu117-cp310-cp310-win_amd64.whl D:/CPS/ai/sd/setup/whl/torchvision-0.14.1+cu117-cp310-cp310-win_amd64.whl"

set TORCH_COMMAND "pip install D:/CPS/ai/sd/setup/whl/torch-2.0.0+cu118-cp310-cp310-win_amd64.whl D:/CPS/ai/sd/setup/whl/torchvision-0.15.0+cu118-cp310-cp310-win_amd64.whl"
pip install D:/CPS/ai/sd/setup/whl/117/torchvision-0.14.1+cu117-cp310-cp310-win_amd64.whl


相关文献