wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-15 16:02:32 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Add amd gpu support

Browse Source 
Testing
This commit is contained in:
LEv145 2023-03-27 09:43:11 +02:00 committed by GitHub
parent a01a151002
commit 55846608bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
docker-compose.yml
View File

@ -15,7 +15,7 @@ x-base_service: &base_service
- ./output:/opt/comfy_ui/output - ./output:/opt/comfy_ui/output
- ./input:/opt/comfy_ui/input - ./input:/opt/comfy_ui/input
- ./custom_nodes:/opt/comfy_ui/custom_nodes - ./custom_nodes:/opt/comfy_ui/custom_nodes
x-nvidia_docker: &nvidia_docker x-nvidia_docker: &nvidia_docker
deploy: deploy:
resources: resources:
@ -25,13 +25,25 @@ x-nvidia_docker: &nvidia_docker
device_ids: ["0"] device_ids: ["0"]
capabilities: [gpu] capabilities: [gpu]
x-base_service_amd: &amd_docker
group_add:
- video
devices:
- "/dev/dri"
- "/dev/kfd"
services: services:
nvidia: nvidia:
<<: *base_service <<: *base_service
<<: *nvidia_docker <<: *nvidia_docker
profiles: ["nvidia"] profiles: ["nvidia"]
amd:
<<: *base_service
<<: *amd_docker
profiles: ["amd"]
cpu: cpu:
<<: *base_service <<: *base_service
profiles: ["cpu"] profiles: ["cpu"]