Update Dockerfile

update Pytorch and add comfy manager
This commit is contained in:
simonmcnair 2025-04-08 15:17:55 +01:00 committed by GitHub
parent 9a9111d640
commit 894b87722b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,4 +1,4 @@
FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
@ -51,6 +51,9 @@ RUN --mount=type=cache,target="$USER_HOME"/.cache/pip \
git reset --hard 276f8fce9f5a80b500947fb5745a4dde9e84622d && \
pip install -r requirements.txt
RUN git clone -b 3.6.5 --depth=1 https://github.com/ltdrdata/ComfyUI-Manager ${ROOT}/custom_nodes/comfyui-manager && \
pip install -r ${ROOT}/custom_nodes/comfyui-manager/requirements.txt
RUN chown -R "$PUID:$PGID" "${ROOT}"
WORKDIR ${ROOT}