当前位置:首页 > 系统教程 > 正文

Ubuntu系统下ComfyUI安装全攻略:开启你的AI图像生成之旅

Ubuntu系统下ComfyUI安装全攻略:开启你的AI图像生成之旅

在人工智能绘画领域,ComfyUI 以其独特的节点式工作流和高效的显存管理脱颖而出。相比于传统的 WebUI,它不仅更具灵活性,还能让用户更深入地理解生成逻辑。本文将为你提供一份详尽的ComfyUI安装教程,带你一步步在 Ubuntu 系统上搭建起属于自己的 AI 创作站。

第一步:基础环境与Ubuntu系统配置

首先,我们需要确保系统环境的纯净与稳定。Ubuntu系统配置的第一步是更新包索引并安装必要的开发工具包。打开终端,执行以下命令:

sudo apt update && sudo apt upgrade -y
sudo apt install git python3-pip python3-venv libgl1 libglib2.0-0 -y

第二步:驱动先行——显卡驱动安装

跑 AI 模型的核心在于 GPU。因此,显卡驱动安装是绝对不能跳过的环节。你可以通过 Ubuntu 的“软件和更新”中的“附加驱动”标签页,选择标注为 (recommended) 的 Nvidia 驱动进行安装。安装完成后,重启电脑并使用 nvidia-smi 命令确认驱动是否生效。

Ubuntu系统下ComfyUI安装全攻略:开启你的AI图像生成之旅 ComfyUI安装教程  Ubuntu系统配置 显卡驱动安装 Stable Diffusion工作流 第1张

第三步:克隆 ComfyUI 并配置虚拟环境

为了避免 Python 依赖冲突,我们强烈建议使用虚拟环境。按照以下步骤拉取源码并初始化:

git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI
python3 -m venv venv
source venv/bin/activate

第四步:安装依赖与模型运行

在虚拟环境中,我们需要安装 PyTorch 以及 ComfyUI 所需的所有第三方库:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt

一切就绪后,将你的模型文件(如 SDXL 或 SD1.5)放入 models/checkpoints 文件夹,执行 python main.py。通过浏览器打开提示的本地地址,你就可以开始构建第一个Stable Diffusion工作流了!

本文核心要点回顾:

  • ComfyUI安装教程:从源码克隆到环境搭建的全流程。
  • Ubuntu系统配置:基础工具链与 Python 虚拟环境的准备。
  • 显卡驱动安装:确保 CUDA 加速正常运行的前提。
  • Stable Diffusion工作流:节点化绘图的高级应用体验。