ComfyUI 本地 AI 绘画工作流：从安装到出图的全新手册

# 1. 克隆仓库（GitHub，慢的话加 ghfast.top 前缀）
git clone https://github.com/comfyanonymous/ComfyUI.git
# 国内：git clone https://ghfast.top/https://github.com/comfyanonymous/ComfyUI.git

# 2. 进入目录
cd ComfyUI

# 3. 创建虚拟环境（推荐，避免和系统 Python 冲突）
python3 -m venv venv

# 4. 激活虚拟环境
source venv/bin/activate    # Linux / macOS / WSL
# Windows CMD: venv\Scripts\activate
# Windows PowerShell: venv\Scripts\Activate.ps1

# 5. 安装依赖
pip install -r requirements.txt
# 国内加清华镜像：pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install comfy-cli
comfy install

#!/usr/bin/env bash
set -u
# ============================================================
#  ComfyUI 本地安装 一键脚本（macOS / Linux / WSL）
#  公开源码，欢迎审查 —— 不放心可先复制给 AI 判断
#  适用场景：NVIDIA GPU + Python 3.12+ 环境下安装 ComfyUI
# ============================================================

GREEN='\033[0;32m'; YELLOW='\033[1;33m'; RED='\033[0;31m'; CYAN='\033[0;36m'; NC='\033[0m'
info()  { echo -e "${GREEN}[INFO]${NC} $1"; }
warn()  { echo -e "${YELLOW}[WARN]${NC} $1"; }
error() { echo -e "${RED}[ERROR]${NC} $1"; }
step()  { echo -e "${CYAN}[STEP]${NC} $1"; }

detect_network() {
  step "检测网络环境（国内 / 国外）..."
  if curl -fsI --max-time 5 "https://claude.ai" >/dev/null 2>&1; then
    info "可直连 claude.ai，判定为海外网络"
    echo "overseas"
  else
    warn "无法直连 claude.ai，判定为国内网络环境（将自动切换镜像源）"
    echo "domestic"
  fi
}

check_gpu() {
  step "检查 GPU 环境..."
  if command -v nvidia-smi >/dev/null 2>&1 && nvidia-smi -L >/dev/null 2>&1; then
    info "检测到 NVIDIA GPU："
    nvidia-smi --query-gpu=name,memory.total --format=csv,noheader 2>/dev/null | head -3
    return 0
  fi
  warn "未检测到 NVIDIA GPU（nvidia-smi 不可用）。ComfyUI 仍可安装，但推理走 CPU 会明显更慢。"
  echo "  · 有独显的笔记本/服务器请先装好 NVIDIA 驱动（命令 nvidia-smi 能输出即 OK）"
  echo "  · Windows 用户请在 WSL 中运行本脚本（NVIDIA 官方支持 WSL CUDA）"
  echo "  · 没有 GPU 也可以继续（CPU 模式），或考虑云端 API 更快"
  return 1
}

check_python() {
  step "检查 Python 3.12+ ..."
  local found=""
  for p in python3.12 python3.13 python3.14 python3; do
    if command -v "$p" >/dev/null 2>&1; then
      local ver
      ver=$("$p" --version 2>&1)
      if echo "$ver" | grep -qE "3\.(1[2-4]|[0-9])"; then
        found="$p"; break
      fi
    fi
  done
  if [ -n "$found" ]; then
    PY="$found"; info "使用 $PY ：$($PY --version 2>&1)"; return 0
  fi
  error "需要 Python 3.12+ 。请先安装（https://www.python.org/downloads/ 或 distro 包管理器）。"
  return 1
}

install_comfyui() {
  local net="$1"
  step "下载 ComfyUI 源码并安装依赖（耗时取决于网速，请耐心等待）..."

  local DIR="comfyui-setup"
  mkdir -p "$DIR" && cd "$DIR" || exit 1

  # --- Step A: 获取源码 ---
  step "获取 ComfyUI 源码 ..."

  if [ "$net" = "domestic" ]; then
    ZIP_URL="https://ghfast.top/https://github.com/comfyanonymous/ComfyUI/archive/refs/heads/master.zip"
  else
    ZIP_URL="https://github.com/comfyanonymous/ComfyUI/archive/refs/heads/master.zip"
  fi

  # 尝试 zip 下载（国内镜像优先）
  if command -v unzip >/dev/null 2>&1; then
    if curl -fsSL --max-time 60 "$ZIP_URL" -o comfyui.zip 2>/dev/null && unzip -q -o comfyui.zip; then
      info "源码下载成功（zip）"
    else
      warn "zip 下载失败，尝试 git clone ..."
    fi
  else
    warn "unzip 未安装，尝试 git clone ..."
  fi

  # 如果 zip 失败了或者没装 unzip，退回到 git
  if [ ! -d "ComfyUI-master" ] && [ ! -d "ComfyUI" ]; then
    if command -v git >/dev/null 2>&1; then
      local GIT_URL
      if [ "$net" = "domestic" ]; then
        GIT_URL="https://ghfast.top/https://github.com/comfyanonymous/ComfyUI.git"
      else
        GIT_URL="https://github.com/comfyanonymous/ComfyUI.git"
      fi
      if git clone --depth 1 "$GIT_URL" 2>/dev/null; then
        info "git clone 成功"
      else
        error "无法获取 ComfyUI 源码。请手动下载 zip 并解压到 comfui-setup/ 目录。"
        echo "  海外用户：https://github.com/comfyanonymous/ComfyUI"
        echo "  国内用户：https://ghfast.top/https://github.com/comfyanonymous/ComfyUI"
        exit 1
      fi
    else
      error "既找不到 unzip 也找不到 git 。请手动下载安装 ComfyUI。"
      exit 1
    fi
  fi

  # 确保进入源码目录（可能是 ComfyUI-master 或 ComfyUI）
  if [ -d "ComfyUI-master" ]; then mv ComfyUI-master ComfyUI; fi
  cd ComfyUI || exit 1

  # --- Step B: 创建虚拟环境 ---
  step "创建 Python 虚拟环境 ..."
  if [ ! -d ".venv" ]; then
    $PY -m venv .venv || { error "创建虚拟环境失败"; exit 1; }
  fi

  # shellcheck disable=SC1091
  source .venv/bin/activate || { warn "激活虚拟环境失败，继续使用系统环境"; }

  # --- Step C: 安装 PyTorch + 依赖 ---
  step "安装 PyTorch（包含 CUDA 支持）..."

  # 根据网络选择 pip 镜像
  local PIP_INDEX="-i https://pypi.org/simple"
  if [ "$net" = "domestic" ]; then
    PIP_INDEX="-i https://pypi.tuna.tsinghua.edu.cn/simple"
    info "使用清华 PyPI 镜像加速"
  fi

  # 安装 PyTorch（含 CUDA 12.x，匹配大多数消费级显卡）
  if pip install torch torchvision torchaudio $PIP_INDEX --timeout 120 >/dev/null 2>&1; then
    info "PyTorch 安装成功"
  else
    warn "默认 CUDA 版本安装失败，尝试 CUDA 12.1 版本 ..."
    if pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 --timeout 120 >/dev/null 2>&1; then
      info "PyTorch CUDA 12.1 安装成功"
    else
      error "PyTorch 安装失败。请确认已安装 NVIDIA 驱动且 CUDA toolkit 版本匹配。"
      exit 1
    fi
  fi

  step "安装 ComfyUI 依赖 ..."
  if pip install -r requirements.txt $PIP_INDEX --timeout 120 >/dev/null 2>&1; then
    info "ComfyUI 依赖安装成功"
  else
    warn "镜像源安装失败，改用官方源重试 ..."
    if pip install -r requirements.txt --timeout 120 >/dev/null 2>&1; then
      info "ComfyUI 依赖安装成功（官方源）"
    else
      error "ComfyUI 依赖安装失败。常见问题："
      echo "  1. 内存不足：请确保至少 8GB 可用内存"
      echo "  2. 网络超时：尝试更换 pip 镜像源后重跑本脚本"
      echo "  3. Python 版本不匹配：要求 3.12+（当前 $(python3 --version 2>&1)"
      exit 1
    fi
  fi

  # --- Step D: 生成启动脚本 ---
  step "生成启动脚本 ..."
  cat > start-comfyui.sh <<'SHEOF'
#!/usr/bin/env bash
cd "$(dirname "$0")"
if [ -f .venv/bin/activate ]; then source .venv/bin/activate; fi
python main.py --preview-method auto
SHEOF
  chmod +x start-comfyui.sh
  info "已生成 start-comfyui.sh 启动脚本"

  # --- Step E: 输出使用说明 ---
  info "============================================"
  info "安装完成！"
  info ""
  info "启动方式（任选其一）："
  info "  cd comfyui-setup/ComfyUI && ./start-comfyui.sh"
  info "  或手动：cd comfyui-setup/ComfyUI && source .venv/bin/activate && python main.py"
  info ""
  info "打开浏览器访问 http://localhost:8188 即可使用 ComfyUI 界面"
  info ""
  info "模型下载提示："
  info "  · 模型文件放到 ComfyUI/models/ 下对应子目录（checkpoints/lora/vae ...）"
  info "  · HuggingFace 国内慢可换 hf-mirror.com 镜像"
  info ""
  info "显卡带不动？考虑云端 API 方案："
  info "  云间中转站 https://cloudzone-api.cyou/ （90+ 模型，OpenAI 兼容）"
  info "============================================"
}

main() {
  echo "============================================"
  echo "  ComfyUI 本地安装 一键脚本"
  echo "  适用于 macOS / Linux / WSL"
  echo "============================================"
  echo ""

  local NET
  NET=$(detect_network)

  check_gpu || warn "继续安装（CPU 模式可跑但慢；想省事可在文末考虑云端 API 路线）"
  if ! check_python; then exit 1; fi

  install_comfyui "$NET"
}

main "$@"

# ============================================================
#  ComfyUI 本地安装 一键脚本（Windows PowerShell 引导器）
#  公开源码，欢迎审查 -- 不放心可先复制给 AI 判断
#  说明：Windows 下推荐使用社区整合包，本脚本提供替代方案。
#        如需 Python 源码安装途径，会通过 WSL 执行。
# ============================================================

$ErrorActionPreference = "Stop"

function Write-Info  { Write-Host "[INFO] $args" -ForegroundColor Green }
function Write-Warn  { Write-Host "[WARN] $args" -ForegroundColor Yellow }
function Write-Err   { Write-Host "[ERROR] $args" -ForegroundColor Red }
function Write-Step  { Write-Host "[STEP] $args" -ForegroundColor Cyan }

# 1. 检测网络环境
Write-Step "检测网络环境（国内 / 国外）..."
try {
    $null = Invoke-WebRequest -Uri "https://claude.ai" -Method Head -TimeoutSec 5 -UseBasicParsing
    $NET = "overseas"
    Write-Info "可直连 claude.ai，判定为海外网络"
} catch {
    $NET = "domestic"
    Write-Warn "无法直连 claude.ai，判定为国内网络环境（将自动使用镜像）"
}

# 2. 检测 GPU
Write-Step "检查 GPU 环境..."
try {
    $gpuInfo = nvidia-smi --query-gpu=name,memory.total --format=csv,noheader 2>$null | Select-Object -First 3
    if ($gpuInfo) {
        Write-Info "检测到 NVIDIA GPU："
        Write-Host $gpuInfo
    } else {
        throw "No GPU detected"
    }
} catch {
    Write-Warn "未检测到 NVIDIA GPU。ComfyUI 仍可安装，但推理走 CPU 会明显更慢。"
    Write-Host "  · 有独显的笔记本/服务器请先装好 NVIDIA 驱动（nvidia-smi 能输出即 OK）"
    Write-Host "  · 没有 GPU 也可以继续（CPU 模式），或考虑用云端 API 更快"
}

# 3. 告知用户：Windows 首选社区整合包（最省心）
Write-Host ""
Write-Host "============================================"
Write-Host "  Windows 用户推荐：使用社区整合包"
Write-Host "============================================"
Write-Host ""
Write-Host "Windows 下最省心的安装方式是直接使用社区提供的整合包，"
Write-Host "解压即用，不需要配置 Python 环境。"
Write-Host ""
Write-Host "安装步骤："
Write-Host "  1. 访问 ComfyUI GitHub（https://github.com/comfyanonymous/ComfyUI ）"
if ($NET -eq "domestic") {
    Write-Host "     （国内：加前缀 ghfast.top/https:// 代理访问）"
}
Write-Host "  2. 进入 Releases 页面，下载 'ComfyUI_windows_portable_*' 压缩包"
Write-Host "  3. 解压到硬盘（路径不要带中文或特殊字符）"
Write-Host "  4. 双击 'run_nvidia_gpu.bat'，等待依赖安装完成"
Write-Host "  5. 浏览器自动打开 http://localhost:8188"
Write-Host ""
Write-Host "注意：整合包由社区维护，非官方发布，文件名和版本号以作者最新发布为准。"
Write-Host "使用前请自行校验文件完整性，风险自担。"
Write-Host ""
Write-Host "============================================"

# 4. 是否要同时获取 WSL 安装版本？
Write-Host ""
$use_wsl = Read-Host "是否在 WSL 中走 Python 源码安装方式？ [Y/N]"

if ($use_wsl -ne "Y" -and $use_wsl -ne "y") {
    Write-Info "好的，只使用整合包方式。后续需要帮助可重新运行此脚本。"
    exit 0
}

# 5. 检查 WSL
Write-Step "检查 WSL 环境..."
if (-not (Get-Command wsl -ErrorAction SilentlyContinue)) {
    Write-Err "未找到 wsl 命令。请先安装 WSL："
    Write-Host "  PowerShell（管理员）执行：wsl --install"
    Write-Host "  重启后安装 Ubuntu 发行版，再运行本脚本。"
    exit 1
}

Write-Host ""
Write-Info "WSL 安装方式将由下方 Linux 脚本处理（见 install-comfyui-guide.sh）。"
Write-Host "你可以："
Write-Host "  1. 复制下方完整脚本源码，在 WSL 中粘贴为 .sh 文件运行"
Write-Host "  2. 或前往 https://cleanresolver.com/scripts/install-comfyui-guide.sh 下载"
Write-Host ""
Write-Info "公开源码，欢迎复制给任何 AI 审查。"
Write-Host ""
Write-Host "============================================"
Write-Info "全部完成！"
Write-Host "============================================"