来源:微信公众号「微技术分享」
原文链接: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 |
| CUDA | cuda-toolkit-12-6 |
| 文生图模型 | 阿里通义 Z-Image-Turbo(6B 参数,8 步出图) |
第一步:NVIDIA 驱动 + CUDA 安装
检查系统中的 NVIDIA 显卡
bashlspci | grep -i nvidia在"附加驱动"中切换 nouveau 驱动,然后重启
重启后安装 580 驱动和 12.6 版本 CUDA
bashwget 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 会报错。
- 如果报错,删除所有安装后重启重来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.safetensorsDiffusion 模型
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.safetensorsText 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。