Add amd gpu support

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

View File

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