Skip to content

Latest commit

 

History

History
112 lines (91 loc) · 3.35 KB

wsl-directml.zh.adoc

File metadata and controls

112 lines (91 loc) · 3.35 KB

在 WSL2 环境下通过 DirectML 运行 ComfyUI

  • AMD/Intel GPU 均可

    • 经测试在 4800H 的核显上能跑（比纯 CPU 快，比 Linux 下 ROCm 慢）

  • 建议使用 Windows 11

  • 建议先更新 WSL2 内核: https://apps.microsoft.com/detail/9P9TQF7MRM4R

    • 或运行 powershell: wsl --update

  • 本文使用 WSL2 openSUSE Tumbleweed: https://apps.microsoft.com/detail/9MSSK2ZXXN11

    • 在微软商店安装后，在开始菜单中找到 openSUSE Tumbleweed 并打开，按提示完成初始配置即可

    • 也可以运行 powershell 命令安装: wsl --install openSUSE-Tumbleweed

1. 安装基础包

Note
 因为 DirectML 支持版本所限，使用 Python 3.10 
sudo zypper install --no-confirm \
    python310 python310-pip \
    python310-wheel python310-setuptools python310-numpy \
    git aria2 \
    Mesa-libGL1 libgthread-2_0-0

(可选) 安装 ComfyUI InstantID 所需包

sudo zypper install --no-confirm \
    python310-devel python310-Cython \
    gcc-c++ cmake

(可选) 设置代理

PIP 代理
sudo pip3.10 config set global.proxy http://host.docker.internal:1081

sudo pip3.10 install --upgrade pip
临时系统代理
export HTTP_PROXY=http://host.docker.internal:1081
export HTTPS_PROXY=$HTTP_PROXY
export http_proxy=$HTTP_PROXY
export https_proxy=$HTTP_PROXY
export NO_PROXY="localhost,*.local,*.internal,[::1],fd00::/7,
10.0.0.0/8,127.0.0.0/8,169.254.0.0/16,172.16.0.0/12,192.168.0.0/16"
export no_proxy=$NO_PROXY

2. 安装 Torch-DirectML (及其所需版本的 PyTorch)

sudo pip3.10 install --break-system-packages \
    torch-directml torchvision torchaudio \
    --index-url https://download.pytorch.org/whl/cpu \
    --extra-index-url https://pypi.org/simple

3. 安装 ComfyUI 所需依赖

sudo pip3.10 install --break-system-packages \
    -r https://raw.githubusercontent.com/comfyanonymous/ComfyUI/master/requirements.txt \
    -r https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/requirements.txt

(可选) 安装 ControlNet Auxiliary Preprocessors 所需依赖

sudo pip3.10 install --break-system-packages \
    -r https://raw.githubusercontent.com/Fannovel16/comfyui_controlnet_aux/main/requirements.txt \
    --extra-index-url https://download.pytorch.org/whl/cpu

(可选) 安装 ComfyUI InstantID 所需依赖

sudo pip3.10 install --break-system-packages \
    -r https://raw.githubusercontent.com/ZHO-ZHO-ZHO/ComfyUI-InstantID/main/requirements.txt

4. 下载 ComfyUI

git clone --depth=1 --no-tags --recurse-submodules --shallow-submodules \
    https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI

5. 下载 ComfyUI Manager

cd custom_nodes
git clone --depth=1 --no-tags --recurse-submodules --shallow-submodules \
    https://github.com/ltdrdata/ComfyUI-Manager.git
cd ..

(可选) 下载一些模型文件

cd models
aria2c https://raw.githubusercontent.com/YanWenKun/ComfyUI-Docker/main/cu121/scripts/download-models.txt
aria2c --allow-overwrite=false --auto-file-renaming=false --continue=true \
    --max-connection-per-server=5 --input-file=download-models.txt
cd ..

6. 运行 ComfyUI

python3.10 main.py --listen --port 8188 --directml
Tip
 如果显存小于 4G，添加参数: --lowvram