Init docker support (Nvidia and AMD)

[LEv145]

Minimum initial docker-compose to run

#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

[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... 😢

[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... 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

[comfyanonymous]

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

[LEv145]

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]

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

[paulo-coronado]

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 :)

[kwaa]

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

[LEv145]

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

[kwaa]

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

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

[LEv145]

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