docker-prompt-generator

环境要求

• 已安装并可用 Docker 环境（官方推荐基于 Docker 与 Nvidia 深度学习基础容器）
• 使用 GPU 镜像时需具备可用的 NVIDIA GPU 及对应容器运行时
• 纯 CPU 环境也可运行，可直接使用 cpu 镜像
• 能访问 GitHub 以克隆项目代码；Docker Hub 可拉取 soulteary/prompt-generator 镜像

安装与启动步骤

1. 1准备 Docker 环境

   按 README 说明先准备好 Docker 环境；GPU 玩法建议参考官方深度学习容器，Mac M1/M2 可参考其 Stable Diffusion 配置文章。

2. 2克隆项目代码

   找一个合适的目录，用 git clone 把 Docker Prompt Generator 项目代码下载到本机（也可直接下载 Zip 压缩包）。

   git clone https://github.com/soulteary/docker-prompt-generator.git

   复制

3. 3启动 CPU 版容器

   无 GPU 或只想用 CPU 时执行此命令，镜像为 cpu 标签，容器端口 7860 映射到宿主机 7860。

   docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -it -p 7860:7860 soulteary/prompt-generator:cpu

   复制

4. 4启动 GPU 版容器

   有可用 GPU 时改用 gpu 标签镜像，其余参数与 CPU 版一致，可借助显卡加速推理。

   docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -it -p 7860:7860 soulteary/prompt-generator:gpu

   复制

5. 5浏览器打开界面

   容器启动后，在浏览器输入运行容器那台主机的 IP 地址加端口 7860，即可开始使用提示词生成工具。

如何确认成功

容器正常运行后，浏览器访问 http://<宿主机IP>:7860 能看到工具界面并可输入提示词，即部署成功。

常见问题

注意事项

• 运行命令中的 --rm 表示容器退出后自动删除，注意不要在容器内保存需要长期保留的数据。
• CPU 与 GPU 镜像都是完整的 docker run 命令，不要删减 --ipc=host、--ulimit 等参数。
• README 未给出生产环境部署、模型下载或账号配置说明，仅适用于本机快速体验。