在 Linux 上安装 ComfyUI

如何在Linux上安装ComfyUI

ComfyUI是一个强大且模块化的Stable Diffusion GUI和后端,它提供了一个基于图/节点/流程图的界面,让你无需编码即可设计和执行高级的Stable Diffusion工作流。以下是在Linux系统上安装ComfyUI的步骤:

1. 克隆仓库

首先,你需要通过Git克隆ComfyUI的仓库到本地。打开终端并运行以下命令:

git clone https://github.com/comfyanonymous/ComfyUI.git

2. 放置模型文件

将你的Stable Diffusion检查点(ckpt/safetensors文件)放置在 models/checkpoints 目录中。

3. 安装PyTorch

根据你的GPU类型安装PyTorch。对于AMD GPU,可以使用以下命令安装PyTorch的稳定版本 或 pytorch nightly:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0
 
# 或者
 
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1

对于NVIDIA GPU,使用以下命令安装:

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
# 或者
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124

4. 安装依赖

在ComfyUI目录中打开终端,然后运行以下命令来安装所有必需的依赖:

pip install -r requirements.txt

5.运行ComfyUI

安装完成后,通过运行以下Python命令启动ComfyUI:

python main.py

6.AMD GPU用户注意事项

如果你使用的是AMD GPU并且不是ROCm官方支持的型号,你可能需要设置环境变量来覆盖GPU版本:

HSA_OVERRIDE_GFX_VERSION=10.3.0 python main.py
# 或者对于RDNA3架构的卡,如7600:
HSA_OVERRIDE_GFX_VERSION=11.0.0 python main.py

7.Intel GPU用户

如果你使用的是Intel GPU,你需要根据Intel的Extension for Pytorch (IPEX)的安装页面安装相应的驱动、Basekit和IPEX包,然后按照Linux的手动安装说明运行ComfyUI。

8.Mac硅芯片用户

如果你使用的是Apple的M1或M2芯片,你需要安装最新的PyTorch夜间版本,并按照Windows和Linux的手动安装说明操作。

遇到问题?

如果在安装过程中遇到任何问题,可以参考ComfyUI的官方文档 (opens in a new tab)或社区论坛获取帮助。