mirror of
https://github.com/okxlin/appstore.git
synced 2025-07-14 05:12:19 +08:00
85 lines
2.7 KiB
Markdown
85 lines
2.7 KiB
Markdown
## 项目介绍
|
||
|
||
stable-deffusion是基于pytorch的目标检测模型，
|
||
当前版本镜像支持GPU，CPU模式需要修改yaml参数，未测试。
|
||
GPU最小8G起，4G显存下未测试
|
||
|
||
安装过程比较漫长，取决于初始模型下载速度，需要耐心等待
|
||
安装面板支持apt、pip指令。详细可参考yml文件，当前版本默认安装即可，由于比较消耗资源，限定镜像仅部署1个
|
||
|
||
本镜像仅做移植，所有功能基于原作者，参考：https://github.com/thirdscam/SD-WebUI-Docker
|
||
|
||
webui有两个可以选择，都在应用商店里，a1111版和comfy
|
||
|
||
安装前确保docker默认支持gpu，配置在/etc/docker/daemon.json上改，像这样：
|
||
```
|
||
{
|
||
"runtimes": {
|
||
"nvidia": {
|
||
"path": "nvidia-container-runtime",
|
||
"runtimeArgs": []
|
||
}
|
||
},
|
||
"default-runtime": "nvidia"
|
||
}
|
||
```
|
||
|
||
- **`1Panel` `v1.10.3-lts`以下版本会覆盖`docker-compose.yml`的`gpu`设置，所以最好安装完成后检查一下，**
|
||
**不对则用以下覆盖并在应用目录下手动执行`docker-compose down && docker-compose up -d`。**
|
||
|
||
```
|
||
version: '3'
|
||
services:
|
||
sd-comfyui:
|
||
build: ./comfy
|
||
image: sd-comfyui:latest
|
||
container_name: ${CONTAINER_NAME}
|
||
restart: always
|
||
runtime: nvidia
|
||
networks:
|
||
- 1panel-network
|
||
ports:
|
||
- "${PANEL_APP_PORT_HTTP}:8188"
|
||
volumes:
|
||
- ${DATA_PATH}:/data
|
||
- ${OUTPUT_PATH}:/output
|
||
environment:
|
||
- CLI_ARGS=${SD_CONSOLE_CLI_COMFY}
|
||
- APT_ARGS=${SD_CONSOLE_APT_COMFY}
|
||
- PIP_ARGS=${SD_CONSOLE_PIP_COMFY}
|
||
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
||
- NVIDIA_VISIBLE_DEVICES=all
|
||
deploy:
|
||
resources:
|
||
reservations:
|
||
devices:
|
||
- driver: nvidia
|
||
device_ids: ['${SD_CONSOLE_DEVICE_IDS}']
|
||
capabilities: [gpu,compute,utility]
|
||
labels:
|
||
createdBy: Apps
|
||
networks:
|
||
1panel-network:
|
||
external: true
|
||
|
||
```
|
||
|
||
## 关于缺少依赖的报错
|
||
安装镜像过程会自动安装当前版本所需依赖
|
||
为适应未来版本变化，添加了apt和pip指令，当前版本不填即可。
|
||
当然如果宿主机未安装nvidia-docker-runtime、nvidia-docker2、nvidia-container-toolkit等nvidia容器运行的依赖，可以使用该功能，只需在APT参数上写包名即可，多个包加空格。
|
||
|
||
|
||
## 关于GPU
|
||
当前仅设置单GPU，nvdia-smi可以查看gpu的ID，默认0，单显卡默认即可
|
||
多显卡可以根据id选择，
|
||
如果需要使用所有显卡，需要更改yaml文件，将id设置为all
|
||
|
||
## 其他
|
||
stable-deffusion相关教程：https://www.bilibili.com/video/BV1As4y127HW/?spm_id_from=333.337.search-card.all.click&vd_source=101386edb3f81944f8492e12ee8cb5b6
|
||
|
||
## api接口
|
||
http://localhost:7860/docs/
|
||
|
||
## 模型下载站
|
||
https://civitai.com/ |