安装必要的依赖项

在Ubuntu系统上安装Stable Diffusion Web UI需要确保系统具备必要的依赖项。运行以下命令更新系统并安装基础工具:

sudo apt update && sudo apt upgrade -y
sudo apt install -y wget git python3 python3-venv python3-pip

NVIDIA显卡用户需安装驱动和CUDA工具包。使用以下命令安装官方驱动:

sudo apt install -y nvidia-driver-525

验证CUDA是否可用:

nvidia-smi

克隆Stable Diffusion Web UI仓库

使用git克隆官方仓库到本地目录。建议选择空间充足的存储位置(至少15GB可用空间):

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui

创建Python虚拟环境

为了避免依赖冲突,建议创建独立的Python虚拟环境:

python3 -m venv venv
source venv/bin/activate

激活环境后提示符前会出现(venv)标记。后续所有操作都需在激活的环境中进行。

安装PyTorch与依赖项

根据CUDA版本安装对应PyTorch。对于CUDA 11.7的用户:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

安装Web UI的其他依赖:

pip install -r requirements.txt

下载模型文件

将Stable Diffusion模型(如v1.5或XL版本)放入指定目录:

mkdir -p models/Stable-diffusion
wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-ckpt.safetensors -O models/Stable-diffusion/v1-5-pruned-ckpt.safetensors

配置启动参数

编辑webui-user.sh文件设置常用参数。典型配置示例:

export COMMANDLINE_ARGS="--listen --enable-insecure-extension-access"

关键参数说明:

  • --listen 允许局域网访问
  • --xformers 启用显存优化(需额外安装)
  • --medvram 中等显存优化(6-8GB显卡)

启动Web UI

运行启动脚本:

./webui.sh

首次启动会自动安装剩余依赖并构建库,耗时可能较长。完成后会显示访问地址(默认http://127.0.0.1:7860)。

故障排查

常见问题解决方案:

内存不足错误:添加--lowvram参数或使用--medvram

./webui.sh --medvram

扩展安装失败:检查网络连接或手动安装

cd extensions
git clone [扩展仓库URL]

性能优化建议

对于NVIDIA显卡,安装xformers可提升20-30%速度:

pip install xformers

启用TensorRT加速需额外步骤:

pip install tensorrt

设置环境变量启用优化:

export TF_ENABLE_ONEDNN_OPTS=1
Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐