Linux部署Stable Diffusion需要哪些步骤

在Linux上部署Stable Diffusion通常有以下步骤：

1. 环境准备：

• 推荐使用Ubuntu 20.04及以上系统，确保有NVIDIA GPU（至少8GB VRAM）、足够内存（16GB+）和存储空间（20GB+）。
• 安装Python 3.8+、pip、Git等基础工具。

1. 安装依赖：

• 安装PyTorch，根据CUDA版本选择合适命令，如pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113（CUDA 11.3为例）。
• 安装Stable Diffusion相关库，如pip install diffusers transformers accelerate。

1. 下载模型：

• 注册Hugging Face账号并同意模型使用条款，安装huggingface_hub，然后通过代码下载模型，如from diffusers import StableDiffusionPipeline; model_id = "CompVis/stable-diffusion-v1-4"; pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)。

1. 运行模型：

• 可通过Python脚本运行，如设置prompt等参数后生成图像并保存。
• 也可使用WebUI方式，如克隆AUTOMATIC1111/stable-diffusion-webui仓库，运行webui.sh脚本启动，通过浏览器访问指定端口使用。