diff --git a/visual-ai/ComfyUI/docker/Dockerfile b/visual-ai/ComfyUI/docker/Dockerfile index bfe4191..3d37b0c 100644 --- a/visual-ai/ComfyUI/docker/Dockerfile +++ b/visual-ai/ComfyUI/docker/Dockerfile @@ -21,7 +21,8 @@ RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRO curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10 && \ update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1 && \ pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/xpu && \ - pip install intel-extension-for-pytorch==2.7.10+xpu oneccl_bind_pt==2.7.0+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \ + pip install intel-extension-for-pytorch==2.7.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \ + pip install oneccl_bind_pt==2.7.0+xpu --index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \ pip install bigdl-core-xe-all==2.6.0 --extra-index-url https://download.pytorch.org/whl/xpu && \ apt remove python3-blinker -y && \ # Install xDit related dependencies diff --git a/visual-ai/Wan2.2/docker/Dockerfile b/visual-ai/Wan2.2/docker/Dockerfile index 2cd5a39..a48e9ee 100644 --- a/visual-ai/Wan2.2/docker/Dockerfile +++ b/visual-ai/Wan2.2/docker/Dockerfile @@ -93,7 +93,8 @@ CMD ["bash", "-c", "source /root/.bashrc && exec bash"] ENV LD_LIBRARY_PATH="/usr/local/lib:/usr/local/lib/python3.10/dist-packages/torch/lib:$LD_LIBRARY_PATH" RUN pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/xpu && \ - pip install intel-extension-for-pytorch==2.7.10+xpu oneccl_bind_pt==2.7.0+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \ + pip install intel-extension-for-pytorch==2.7.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \ + pip install oneccl_bind_pt==2.7.0+xpu --index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \ pip install bigdl-core-xe-all==2.6.0 --extra-index-url https://download.pytorch.org/whl/xpu && \ apt remove python3-blinker -y