wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-06 07:12:30 +08:00
Code Issues Actions 26 Packages Projects Releases Wiki Activity

Add dockerfile for rocm

Browse Source
This commit is contained in:
Vicky Tsang 2026-04-02 12:32:05 -04:00
parent 0c63b4f6e3
commit 457c6de393
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
docker/Dockerfile.rocm Normal file
View File

@ -0,0 +1,18 @@
FROM rocm/primus:v26.2
ENV COMFYUI_PATH=/workload/ComfyUI
ENV COMFYUI_BRANCH=amd-integration
RUN git clone --branch $COMFYUI_BRANCH https://github.com/rocm/comfyui.git $COMFYUI_PATH && \
git clone https://github.com/ltdrdata/ComfyUI-Manager $COMFYUI_PATH/custom_nodes/comfyui-manager && \
cd $COMFYUI_PATH/custom_nodes/comfyui-manager && \
git checkout 4.1 && \
git clone https://github.com/rgthree/rgthree-comfy.git $COMFYUI_PATH/custom_nodes/rgthree-comfy && \
cd $COMFYUI_PATH/custom_nodes/rgthree-comfy && \
git checkout 8ff50e4
RUN pip install --upgrade pip
RUN sed -i 's/torch/#torch/g' $COMFYUI_PATH/requirements.txt
RUN pip install -r $COMFYUI_PATH/requirements.txt && \
pip install torchsde && \
pip install -r $COMFYUI_PATH/custom_nodes/comfyui-manager/requirements.txt
CMD ["/bin/bash"]