安装 WSL
在管理员 PowerShell 输入命令 wsl --install
,之后终端会默认安装 Ubuntu。
安装 Docker Desktop
安装后点击左侧的 Add Extensions,推荐 Disk usage 扩展,便于管理 Docker 存储空间。
下载 Stable Diffusion WebUI Docker
选择最新版本下载并解压到本地硬盘。
检查WIN10的CUDA
启动ubuntu后输入nvidia-smi
命令,win10会报
Failed to initialize NVML: GPU access blocked by the operating system
Failed to properly shut down NVML: GPU access blocked by the operating system
因为win10版本是19043,升级到19044解决,使用易升进行自动升级。
如果显示CUDA版本请继续。
选择分支
目前 Stable Diffusion 有 hlky、auto、auto-cpu 和 lstein 四个分支。如果要更换分支,则更改镜像构建命令 docker compose --profile [ui] up --build,将 [ui] 替换为所需的镜像名即可。
- hlky(推荐):界面直观,最高分辨率为 1024x1024,是最受欢迎的主题,镜像构建命令为
docker compose --profile hlky up --build
。 - auto:设置模块最丰富,显示绘画过程,支持随机插入艺术家、参数读取和否定描述,最高分辨率为 2048x2048(高分辨率对显存要求更高),镜像构建命令为
docker compose --profile auto up --build
。 - auto-cpu:唯一不依赖显卡的分支。如果没有符合要求的显卡,可以使用 CPU 版本,稍后的镜像构建命令为
docker compose --profile auto-cpu up --build
。 - lstein:cli 端非常成熟,WebUI 端参数较少,能自动读取图片记录,适合无进阶需求的新手和 Mac 用户使用,镜像构建命令为
docker compose --profile lstein up --build
。
启动 Stable Diffusion
例如解压位置是C:\stable-diffusion-webui-docker-2.0.1,在ubuntu上执行命令:
cd /mnt/c/stable-diffusion-webui-docker-2.0.1
执行命令自动下载模型和依赖包:
docker compose --profile download up --build
继续执行构建(注意选择需要的分支):
docker compose --profile hlky up --build
构建完成后,提示访问 http://localhost:7860/
,你就可以在本地 AI 生成图片了。