ComfyUI icon indicating copy to clipboard operation
ComfyUI copied to clipboard

Published 20 hours ago • verified publisher icon comfyanonymous

Init docker support (Nvidia and AMD)

Open LEv145 opened this issue 2 years ago • 10 comments

Minimum initial docker-compose to run изображение

https://github.com/comfyanonymous/ComfyUI/issues/188

How to use:

docker compose --profile cpu up  # CPU
docker compose --profile nvidia up  # Nvidia with https://download.pytorch.org/whl/cu117 
docker compose --profile amd up  # AMD with https://download.pytorch.org/whl/rocm5.4.2
docker compose --profile amd6600 up  # like amd profile, but with HSA_OVERRIDE_GFX_VERSION=10.3.0

LEv145 avatar Mar 20 '23 23:03 LEv145

Hey @LEv145! Thanks for sharing it! I am also working with Comfy + Docker hosted on Runpod's Serverless GPU, but I am getting NVIDIA erros. I can't link my GPU with the containers... 😢

paulo-coronado avatar Mar 24 '23 17:03 paulo-coronado

Hey @LEv145! Thanks for sharing it! I am also working with Comfy + Docker hosted on Runpod's Serverless GPU, but I am getting NVIDIA erros. I can't link my GPU with the containers... cry

I'm happy to share On my graphics card, docker-compose works stably without errors

Probably a problem in nvidia-docker, I hope you fix it

LEv145 avatar Mar 24 '23 20:03 LEv145

Unfortunately I don't have a nvidia card so I can't test this.

comfyanonymous avatar Mar 27 '23 06:03 comfyanonymous

Unfortunately I don't have a nvidia card so I can't test this.

I have nvidia 2600 super working on my card, it should work on any card from nvidia, because I use nvidia-docker

I can add a version without gpu to docker

LEv145 avatar Mar 27 '23 06:03 LEv145

I'm edited docker architecture

Usage:

cd docker
docker compose --profile cpu up  # CPU
docker compose --profile nvidia up  # Nvidia with https://download.pytorch.org/whl/cu117 
docker compose --profile amd up  # AMD with https://download.pytorch.org/whl/rocm5.4.2
docker compose --profile amd6600 up  # like amd profile, but with HSA_OVERRIDE_GFX_VERSION=10.3.0

LEv145 avatar Mar 27 '23 15:03 LEv145

Hello, @LEv145! Thanks for adding Docker support! Could you please add me on Discord (@pcoronado#4338)? I am new in docker compose and I have some questions about your fork :)

paulo-coronado avatar Mar 27 '23 16:03 paulo-coronado

If merged, I would be interested in adding Intel Arc Graphics support.

kwaa avatar Apr 24 '23 05:04 kwaa

Good stuff: https://github.com/AbdBarho/stable-diffusion-webui-docker/pull/384

LEv145 avatar Apr 28 '23 10:04 LEv145

Good stuff: AbdBarho/stable-diffusion-webui-docker#384

Unfortunately, this repo only seems to support NVIDIA graphics cards.

kwaa avatar Apr 28 '23 11:04 kwaa

https://github.com/YanWenKun/ComfyUI-Docker

LEv145 avatar Jun 01 '23 22:06 LEv145