You can either build this Docker image yourself, or alternatively, you can use one of my pre-built images:
| CUDA Version | Torch | xformers | Image |
|---|---|---|---|
| 12.4 | 2.6.0 | 0.0.29.post3 | ghcr.io/ashleykleynhans/runpod-worker-comfyui:4.0.2-cuda12.4.1 |
| 12.8 | 2.10.0 | 0.0.34 | ghcr.io/ashleykleynhans/runpod-worker-comfyui:4.0.2-cuda12.8.1 |
If you choose to build it yourself:
- Sign up for a Docker hub account if you don't already have one.
- Build the Docker image on your local machine and push to Docker hub:
# Clone the repo
git clone https://github.com/ashleykleynhans/runpod-worker-comfyui.git
cd runpod-worker-comfyui
# Build and push
docker build -t dockerhub-username/runpod-worker-comfyui:1.0.0 .
docker login
docker push dockerhub-username/runpod-worker-comfyui:1.0.0If you're building on an M1 or M2 Mac, there will be an architecture
mismatch because they are arm64, but Runpod runs on amd64
architecture, so you will have to add the --plaform as follows:
docker buildx build --push -t dockerhub-username/runpod-worker-comfyui:1.0.0 . --platform linux/amd64