mirror of
https://github.com/okxlin/appstore.git
synced 2025-07-13 21:02:18 +08:00
2.7 KiB
2.7 KiB
项目介绍
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