25 lines
1.0 KiB
YAML
25 lines
1.0 KiB
YAML
services:
|
|
comfyui:
|
|
container_name: ai-comfyui
|
|
build:
|
|
context: .
|
|
dockerfile: dockerfile-comfyui
|
|
args:
|
|
ComfyUI_Release: "0.3.43" # Must be (Maj.Min.Patch) format. -> https://github.com/comfyanonymous/ComfyUI/releases
|
|
Python_Release: "3.12.11" # Must be (Maj.Min.Patch) format. -> https://www.python.org/downloads/source/
|
|
Pytorch_Release: "2.7.1" # Must be (Maj.Min.Patch) format. -> # https://github.com/pytorch/pytorch/releases
|
|
Pytorch_Jobs: 12 # Set the number of jobs for the Pytorch_Build -> (Gb of RAM)/17 rounded down to the nearest whole number
|
|
CUDA_ARCH_LIST: "8.9" # Set the CUDA arch to compile for -> get the current cuda version from 'nvidia-smi'
|
|
volumes:
|
|
- ./data/comfyui/models:/app/comfyui/models
|
|
- ./data/comfyui/output:/app/comfyui/output
|
|
ports:
|
|
- 8188:8188
|
|
restart: always
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: [gpu] |