Skip to content

来源:微信公众号「微技术分享」
原文链接:https://mp.weixin.qq.com/s/PhoakvSzf7644k3ay19FBQ

背景

书接上文《给自己搭一套本地AI大模型》,投入 300 元入手 Tesla M60 显卡,之前用来跑本地大模型实现 token 自由。这次继续压榨这张卡的剩余价值——用 ComfyUI 搭建本地 AI 文生图。

环境配置

项目详情
操作系统Ubuntu 24
内存16GB + 16GB 缓存
显卡Tesla M60(8GB 显存)
NVIDIA 驱动nvidia-driver-580
CUDAcuda-toolkit-12-6
文生图模型阿里通义 Z-Image-Turbo(6B 参数,8 步出图)

第一步:NVIDIA 驱动 + CUDA 安装

  1. 检查系统中的 NVIDIA 显卡

    bash
    lspci | grep -i nvidia
  2. 在"附加驱动"中切换 nouveau 驱动,然后重启

  3. 重启后安装 580 驱动和 12.6 版本 CUDA

    bash
    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
    sudo dpkg -i cuda-keyring_1.1-1_all.deb
    sudo apt-get update
    sudo apt install nvidia-driver-580
    nvidia-smi  # 验证驱动
    sudo apt install cuda-toolkit-12-6
    nvcc --version  # 验证 CUDA

⚠️ 注意:老卡只能装 CUDA 12 版本,否则 ComfyUI 会报错。

  1. 如果报错,删除所有安装后重启重来
    bash
    sudo apt --purge remove nvidia*
    sudo apt --purge remove "cublas" "cuda*"
    sudo apt autoremove

第二步:安装 uv 工具和 Python 环境

bash
curl -LsSf https://astral.sh/uv/install.sh | sh
uv python install 3.12

第三步:安装 ComfyUI

拉取源码 + 创建虚拟环境

bash
git clone https://github.com/Comfy-Org/ComfyUI.git
cd ComfyUI && uv venv --python=3.12
source .venv/bin/activate

安装依赖

bash
uv pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

替换 torch 组件版本(适配 CUDA 12.6)

bash
uv pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

启动 ComfyUI

bash
python3 main.py --listen 0.0.0.0 --disable-cuda-malloc

第四步:下载必要模型

VAE 模型

bash
cd models/vae
wget https://www.modelscope.cn/models/Comfy-Org/z_image_turbo/resolve/master/split_files/vae/ae.safetensors

Diffusion 模型

bash
cd models/diffusion_models
wget https://hf-mirror.com/T5B/Z-Image-Turbo-FP8/resolve/main/z-image-turbo-fp8-e4m3fn.safetensors
wget https://modelscope.cn/models/Comfy-Org/z_image_turbo/resolve/main/split_files/diffusion_models/z_image_turbo_bf16.safetensors

Text Encoder

bash
cd models/text_encoders
wget https://modelscope.cn/models/Comfy-Org/z_image_turbo/resolve/main/split_files/text_encoders/qwen_3_4b.safetensors

第五步:测试验证

重新启动 ComfyUI:

bash
source .venv/bin/activate
python3 main.py --listen 0.0.0.0 --disable-cuda-malloc

使用阿里通义 Z-Image-Turbo 模型,仅 6B 参数,8 步即可高效出图。

示例提示词

《新世纪福音战士》明日香 cos,鹅蛋脸,蓝眼睛,穿着校服,微笑,眨眼,低头但眼睛看向观众,手指指向观众。写实风格,特写,环境光,摄像机从上往下俯拍,教室背景。

总结

全程看似简单,实则一点也不难。300 元的 Tesla M60(8G 显存)+ Ubuntu 24 + ComfyUI + Z-Image-Turbo,就能搭建一套本地 AI 文生图工作站,完全离线运行,不用花钱调 API。