为什么要搭建本地 AI 基础设施?
2026 年,AI 工具已经成为数字创作者、开发者和企业的基础设施。但大多数人仍然依赖云端 API — 每次呼叫花钱、每次传输冒隐私风险、每次被速率限制卡住脖子。
本地 AI 基础设施给你三样东西:完全的控制权、无限制的使用量、零边际成本。
一旦搭建完成,你可以:
- 无限制使用 LLM,不用担心 API 费用
- 处理敏感数据,不用离开你的机器
- 同时运行多个服务,没有速率限制
- 与 Cowork MCP 等框架整合,打造真正的 AI 代理团队
这份指南假设你已经有一台带有 GPU 的机器(推荐 DGX GB10、RTX 4090 或更好)。我会从零开始,一步步带你搭建完整的本地 AI 环境。
你今天会学到:
- 安装和配置本地 LLM 服务(vLLM)
- 部署 ComfyUI 图像生成管线
- 管理 GPU 内存和服务启动顺序
- 自动化服务管理(systemd)
- 与 Cowork MCP 整合,打造多 Agent 系统
我们开始吧。
前置需求
开始之前,请确认你已准备好:
| 项目 | 版本需求 | 检查指令 |
|---|---|---|
| Linux | Ubuntu 22.04+ 或 Arch | uname -a |
| GPU | NVIDIA RTX 3090+/4090/GB10 | nvidia-smi |
| CUDA | ≥ 12.4 | nvcc --version |
| Docker | ≥ 24.0 | docker --version |
| Node.js | ≥ 20 | node --version |
| Python | ≥ 3.10 | python3 --version |
如果你的 GPU 内存不足 24GB,建议关闭不必要的图形界面或服务再开始。
步骤 1:安装 Docker 和 NVIDIA Container Toolkit
Docker 是部署本地 AI 服务的推荐方式,因为它能隔离依赖、版本冲突和系统污染。
# 安装 Docker
sudo apt update
sudo apt install -y docker.io docker-compose-v2
# 安装 NVIDIA Container Toolkit(让 Docker 可以使用 GPU)
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt update
sudo apt install -y nvidia-container-toolkit
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker
验证 GPU 可用性:
docker run --rm --gpus all nvidia/cuda:12.4.0-base-ubuntu22.04 nvidia-smi
如果看到 GPU 信息和内存使用率,就代表环境设定正确。
步骤 2:部署 vLLM 本地语言模型
vLLM 是当前最高效的本地 LLM 推理框架,支持多种模型格式(GGUF、GPTQ、AWQ)。
2.1 使用 Docker 部署
docker run -d \
--name vllm-server \
--gpus all \
-p 8000:8000 \
-v ~/.cache/huggingface:/root/.cache/huggingface \
vllm/vllm-openai:latest \
--model "nvidia/Llama-3.1-Nemotron-70B-Instruct" \
--gpu-memory-utilization 0.9 \
--max-model-len 8192
参数说明:
--model: 要部署的模型(可替换为任何 HuggingFace 模型)--gpu-memory-utilization: 使用 GPU 内存的比例(0.9 表示 90%)--max-model-len: 最大上下文长度
注意: 如果你的 GPU 内存小于 80GB,建议从较小的模型开始(7B 或 13B),然后再尝试更大的模型。
2.2 测试 API 端点
curl -X POST http://localhost:8000/v1/completions \
-H "Content-Type: application/json" \
-d '{
"model": "nvidia/Llama-3.1-Nemotron-70B-Instruct",
"prompt": "解释什么是多 Agent AI 系统",
"max_tokens": 200
}'
如果收到回应,vLLM 已经成功部署。
2.3 设定为系统服务(自动启动)
创建 systemd 服务文件:
sudo tee /etc/systemd/system/vllm.service << 'EOF'
[Unit]
Description=vLLM Local LLM Server
After=docker.service
Requires=docker.service
[Service]
Type=oneshot
RemainAfterExit=yes
ExecStart=/usr/bin/docker start vllm-server
ExecStop=/usr/bin/docker stop vllm-server
[Install]
WantedBy=multi-user.target
EOF
启用并启动服务:
sudo systemctl daemon-reload
sudo systemctl enable vllm.service
sudo systemctl start vllm.service
步骤 3:部署 ComfyUI 图像生成
ComfyUI 是当前最强大的本地图像生成工具,支持 Flux、Stable Diffusion、LTX 等各种模型。
3.1 使用 Docker 部署
docker run -d \
--name comfyui \
--gpus all \
-p 8188:8188 \
-v ~/.cache/huggingface:/root/.cache/huggingface \
-v ~/comfyui-output:/output \
ghcr.io/ai-forest/comfyui:latest
这会在 http://localhost:8188 启动 ComfyUI。
3.2 手动安装(进阶选项)
如果你需要更多控制,可以选择手动安装:
# 克隆 ComfyUI 仓库
cd ~/comfyui
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI
# 安装依赖
pip install -r requirements.txt
# 安装自定义节点(可选)
git clone https://github.com/Fannovel16/ComfyUI-Frame-Interpolation.git custom_nodes/
git clone https://github.com/laksjdjoy/deforum-comfy.git custom_nodes/
# 启动服务
python main.py --listen 0.0.0.0 --port 8188
3.3 常用模型下载
# 使用 HuggingFace CLI 下载模型
pip install huggingface_hub
huggingface-cli download stabilityai/stable-diffusion-xl-base-1.0 --local-dir ~/models/sdxl
huggingface-cli download stabilityai/stable-diffusion-3.5-large --local-dir ~/models/sd3.5
步骤 4:GPU 内存管理策略
在本地部署 AI 服务时,GPU 内存是最大的瓶颈。以下是几个实战经验:
4.1 检查可用内存
free -h
nvidia-smi --query-gpu=memory.used,memory.free --format=csv
4.2 服务启动顺序建议
- 先启动 vLLM(语言模型通常内存需求较固定)
- 后启动 ComfyUI(图像生成可以动态调整内存使用)
- 关闭不必要的图形应用(浏览器、Discord 等)
4.3 内存监控脚本
创建一个简单的监控脚本:
#!/bin/bash
# monitor-gpu.sh
while true; do
echo "=== $(date) ==="
nvidia-smi --query-gpu=memory.used,memory.free,temperature.gpu --format=csv
docker ps --filter "name=vllm\|name=comfyui" --format "table {{.Names}}\t{{.Status}}"
sleep 30
done
步骤 5:整合 Cowork MCP 多 Agent 框架
一旦本地 AI 基础设施就绪,就可以连接 Cowork MCP 打造多 Agent 系统。
5.1 注册 vLLM 作为大脑
# 如果使用 Hermes Agent
hermes register-cowork
# 或者手动设定 MCP 端点
# 在你的 Agent 设定档中添加:
{
"mcpServers": {
"cowork": {
"url": "http://localhost:6868/mcp",
"transport": "streamable-http"
},
"vllm": {
"url": "http://localhost:8000/v1",
"transport": "streamable-http"
}
}
}
5.2 实际应用:本地 AI 内容制作管线
本地 vLLM (内容生成) → ComfyUI (图像生成) → ffmpeg (后处理)
这个管线的每一个步骤都可以由不同的 AI 代理协调:
- 研究代理:使用 vLLM 生成内容大纲
- 写作代理:使用 vLLM 扩充为完整文章
- 图像代理:使用 ComfyUI 生成对应插图
- 整合代理:将内容和图像组合成最终输出
步骤 6:自动化与备份
6.1 容器备份
# 备份 ComfyUI 容器状态
docker commit comfyui comfyui-backup:$(date +%Y%m%d)
# 导出镜像
docker save comfyui:latest | gzip > ~/backups/comfyui-$(date +%Y%m%d).tar.gz
6.2 模型备份
# 定期备份 HuggingFace 缓存
rsync -avh ~/.cache/huggingface ~/backups/huggingface-cache/
# 或使用 rclone 备份到云端
rclone sync ~/.cache/huggingface remote:backup-huggingface --progress
6.3 健康检查脚本
创建健康检查 cron job:
# 每 5 分钟检查一次服务状态
*/5 * * * * curl -f http://localhost:8000/v1/models || echo "vLLM is down at $(date)" | mail -s "vLLM Alert" admin@example.com
常见问题排查
Q1: Docker 无法使用 GPU
# 检查 NVIDIA Container Toolkit 是否安装
nvidia-ctk runtime verify
# 重启 Docker 服务
sudo systemctl restart docker
Q2: GPU 内存不足
# 降低 vLLM 内存使用率
# 在 docker run 指令中加入:
--gpu-memory-utilization 0.7
# 或关闭不必要的服务
sudo systemctl stop bluetooth
Q3: ComfyUI 启动失败
# 检查 Docker 日志
docker logs comfyui
# 尝试重新启动
docker restart comfyui
Q4: 模型下载太慢
# 使用 HuggingFace 镜像站(中国大陆用户)
export HF_ENDPOINT=https://hf-mirror.com
下一步
你现在拥有了一套完整的本地 AI 基础设施:
今天:
- 确认 vLLM 和 ComfyUI 都能正常运行
- 测试 API 端点,确认可以生成内容和图像
本周:
- 设定 systemd 服务确保开机自动启动
- 建立备份和监控机制
- 连接至少一个 LLM 后端到 Cowork MCP
本月:
- 探索 Cowork MCP 框架,打造更多 AI 代理团队
- 尝试不同的模型,找出最适合你需求的配置
- 建立自己的 AI 内容制作管线
常见问题解答
Q:我需要什么等级的 GPU 才能运行这些服务? A:建议至少 24GB 内存(RTX 3090/4090)。如果内存较小,可以运行较小的模型(7B-13B)或使用 Docker 的内存限制。
Q:可以在没有 GPU 的机器上运行吗? A:可以,但速度会很慢。CPU 推理是可行的,但对于大型模型(70B+)不实际。
Q:如何更新模型?
A:使用 HuggingFace CLI:huggingface-cli download <model-name> --local-dir <path>。Docker 容器会自动使用更新的本地模型。
Q:vLLM 和 Ollama 有什么不同? A:vLLM 是针对高吞吐量推理优化的框架,支持更多模型格式和进阶功能。Ollama 更轻量、易于使用,但功能较少。对于本地 AI 基础设施,建议使用 vLLM。
Q:ComfyUI 和 Automatic1111 有什么区别? A:ComfyUI 采用节点式工作流程,更灵活、更高效。Automatic1111 有图形界面,对初学者更友好。两个都可以很好地工作,但 ComfyUI 在生产环境中更受欢迎。
阅读时间:约 15 分钟 | 发布日期:2026-07-27
觉得这份指南有帮助?探索 Slashman Tools 上的更多指南。如果你想快速部署 AI 基础设施,查看我们的 AI 工具链模板库 获取即拿即用的设定档。
[- 返回首页]