nicklucche/stable-diffusion

Stability.AI Stable Diffusion 镜像文档

镜像概述

本Docker镜像专为GPU环境设计，用于运行Stability.AI的Stable Diffusion模型，并集成了简单的Web界面，简化AI图像生成的部署和使用流程。

核心功能与特性

• GPU优化：针对NVIDIA GPU进行优化，需配合CUDA环境运行
• 模型集成：内置Stability.AI官方Stable Diffusion模型
• Web界面：提供直观的网页操作界面，无需命令行交互
• 快速部署：预配置环境，开箱即可使用

使用场景与适用范围

• 个人AI图像创作：通过Web界面快速生成图像
• 小型团队原型开发：快速搭建Stable Diffusion服务进行测试
• 教育与演示：展示AI图像生成技术的部署与应用

使用方法与配置说明

前置要求

• 安装Docker Engine (20.10+)
• 安装NVIDIA GPU驱动及nvidia-docker运行时
• 具备支持CUDA的NVIDIA GPU

基本运行命令

bash
docker run --gpus all -p 7860:7860 [镜像名称]

说明：--gpus all 表示使用所有可用GPU；-p 7860:7860 映射Web界面端口

访问Web界面

容器启动后，通过浏览器访问以下地址打开Web界面：

http://localhost:7860

高级配置选项

自定义端口映射

bash
docker run --gpus all -p 8080:7860 [镜像名称]

将主机8080端口映射到容器7860端口，通过http://localhost:8080访问

挂载本地模型目录

bash
docker run --gpus all -p 7860:7860 -v /本地模型路径:/app/models [镜像名称]

挂载本地自定义模型，供Stable Diffusion使用

设置环境变量

bash
docker run --gpus all -p 7860:7860 -e MODEL_PATH=/app/models/custom_model [镜像名称]

通过MODEL_PATH环境变量指定自定义模型路径