本地搭建StableDiffusion服务器


使用版本 : AUTOMATIC1111

系统 : windows10 64位

显卡 : GTX1050Ti

显卡驱动版本 : 516.94

CUDA 版本 : 11.7

显存 : 理论上至少需要4GB以上, 但我只有1GB显存, 但是在修改启动脚本参数后也能够勉强运行, 方法之后附上

依赖程序 :

  1. Git 官网下载安装即可, 版本不重要, 记得添加到环境变量中
  2. Python 建议使用3.10.9版本或更新版本, 下载地址 : https://mirrors.huaweicloud.com/python/3.10.9/python-3.10.9-amd64.exe, 记得需要将python添加到环境变量中
  3. Nvidia CUDA 这里可以查看显卡驱动对应的CUDA版本 : https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 这里可以下载各版本的CUDA : https://developer.nvidia.com/cuda-toolkit

网络代理(可选):

https://ghproxy.com/https://github.com/docmirror/dev-sidecar/releases/download/v1.7.3/DevSidecar-1.7.3.exe

安装完毕后记得安装根证书, 点击 pip加速,选择使用aliyun镜像

服务器搭建流程(默认已经安装完成各种依赖程序)
  1. 更新pip
    1
    pip install --upgrade pip
  2. PyTorch 安装 因为我用的是CUDA11.7, 所以链接地址是11.7的下载地址, 如果你用的是其他版本, 需要修改为对应的下载地址
    1
    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
  3. 下载 StableDiffusionWebUI 代码
    1
    git clone https://ghproxy.com/https://github.com/AUTOMATIC1111/stable-diffusion-webui.git stable-diffusion-webui
  4. 编辑launch.py 这个python文件中会使用到git, 国内git访问速度较慢, 添加代理, 具体流程如下:
    搜索https://github.com/, 替换为https://ghproxy.com/https://github.com/
  5. 创建python环境
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    cd stable-diffusion-webui
    rem 创建环境
    python -m venv .\venv
    cd .\venv\Script
    rem 激活环境
    .\activate.bat
    rem 回到项目根目录
    cd ..\..\
    rem 安装依赖库, 可能需要很久
    pip install -r requirements_versions.txt
  6. 编译安装xformers(可选)
    1
    2
    3
    4
    5
    git clone https://github.com/facebookresearch/xformers.git repositories/xformers
    cd repositories/xformers
    git submodule update --init --recursive
    pip install -r requirements.txt
    pip install -e .
  7. 修改webui-user.bat
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    @echo off

    :: %medvram 6GB及以上显存%
    :: %lowvram 4GB以上%
    :: %lowram 16GB以上%
    :: %skip-torch-cuda-test 去掉显卡内存检查%
    :: xformers 启用xformers
    :: deepdanbooru 插件
    :: precision full 最高精度
    :: no-half 不要将模型切换到16位浮点数

    set PYTHON=
    set GIT=
    set VENV_DIR=venv
    :: 如果你的显卡是4GB COMMANDLINE_ARGS最后添加 --lowvram 6GB就是--medvram 16GB--lowram
    :: 如果有安装xformers, 可以添加--xformers
    :: 显卡满足条件的(4GB及以上)可以去掉--skip-torch-cuda-test
    :: 如果你像我一样只有1G显卡, 这里就像我一样配置就可以了
    set COMMANDLINE_ARGS=--precision full --no-half --skip-torch-cuda-test --always-batch-cond-uncond

    call webui.bat
  8. 启动 命令行输入webui-user.bat即可启动
  9. 测试 浏览器输入http://127.0.0.1:7860/ 因为显卡不够, 所以生成的时候需要注意参数设置, width和heigh尽量不要超过512, 以下是测试使用的prompt :
    (masterpiece : 1.331),
    best quality,
    cartoon,
    (a girl : 1.331),
    double ponytail,
    long bangs,
    big eyes,
    black eyes,
    black hair,
    skirt,
    standing under a big tree,
    flowers all over the floor
    生成的图片就不放出来了.
    每个条件用逗号分隔, 详细用法自行百度, 描述应当准确详细, 生成的图片质量才会好一点
  10. 下载模型包 模型包才是stablediffusion的灵魂, 下载地址 :
    https://huggingface.co/models?sort=downloads
    将下载的ckpt文件或者safetensors格式包放到./models/Stable-diffusion/目录下即可
    1G显存实在不够用, 很多模型包都没法使用, 限制太多, 以下几个模型包测试过基本可用:
    models–dreamlike-art–dreamlike-photoreal-2.0
    momoko-any3.0.vae.pt
    v1-5-pruned-emaonly.safetensors