ComfyUI/docker/docker-compose.yml
LEv145 26ef6d9061 Update docker
Add path for docker
Add Dockerfile for AMD
2023-03-27 16:39:58 +02:00

75 lines
1.2 KiB
YAML

version: "3.9"
x-base_service: &base_service
image: comfyui:1
build:
context: docker_files
dockerfile: default.Dockerfile
environment:
- CLI_ARGS=--listen --port 8188
ports:
- "8188:8188"
volumes:
- models:/opt/comfy_ui/models
- output:/opt/comfy_ui/output
- input:/opt/comfy_ui/input
- custom_nodes:/opt/comfy_ui/custom_nodes
restart: unless-stopped
x-nvidia_docker: &nvidia_docker
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ["0"]
capabilities: [gpu]
x-base_service_amd: &amd_docker
build:
context: docker_files
dockerfile: amd.Dockerfile
group_add:
- video
devices:
- "/dev/dri"
- "/dev/kfd"
services:
nvidia:
<<: *base_service
<<: *nvidia_docker
profiles: ["nvidia"]
amd:
<<: *base_service
<<: *amd_docker
profiles: ["amd"]
amd6600:
<<: *base_service
<<: *amd_docker
profiles: ["amd6600"]
environment:
- HSA_OVERRIDE_GFX_VERSION=10.3.0
- CLI_ARGS=--listen --port 8188
cpu:
<<: *base_service
profiles: ["cpu"]
environment:
- CLI_ARGS=--listen --port 8188 --cpu
volumes:
models:
output:
input:
custom_nodes: