GPU节点管理服务器

一个本地Python MCP服务器,用于通过ZeroTier和SSH管理家庭实验室中的GPU节点,支持ComfyUI、Ollama等多种AI服务。

  • by DavidJBarnes
  • Python
  • 本地部署
  • 更新于
在 GitHub 查看 DavidJBarnes/mcp-server

使用说明

mcp服务器

本地Python MCP服务器,用于通过ZeroTier通过SSH管理家庭实验室GPU节点。

主机

主机 GPU 服务
3090.zero RTX 3090(24GB) ComfyUI(8188)、A1111(7860)
2070.zero RTX 2070(8GB) 奥利马(11434)
dns.zero Raspberry Pi 5 DNS

先决条件

  • Python 3.11+
  • uv 已安装
  • SSH密钥访问 .zero 主机通过ZeroTier
  • ZeroTier在本地运行 .zero 可访问的主机

设置

cd ~/projects/mcp-server
uv sync

使用克劳德代码注册

claude mcp add --scope user mcp-server \
  python3 ~/projects/mcp-server/server.py

验证

claude mcp list

运行测试

uv run pytest

工具参考

工具 目标主机 描述
gpu_status 3090/2070 GPU VRAM、临时、利用率和进程
system_load 任何 CPU、RAM和磁盘使用情况
kill_process 3090/2070 通过PID终止进程
comfyui_queue_status 3090 ComfyUI队列正在运行/挂起作业
comfyui_submit_workflow 3090 提交ComfyUI工作流
comfyui_cancel_job 3090 取消排队的ComfyUI作业
comfyui_list_loras 3090 列出ComfyUI LoRA型号
comfyui_get_output_images 3090 获取提示的输出图像
ollama_list_models 2070 列出Olama车型
ollama_generate 2070 生成文本/视觉完成
ollama_ps 2070 列出加载的Olama型号
caption_image 2070 通过视觉模型为图像添加标题
generate_video_prompt 2070 将字幕转换为视频提示
a1111_status 3090 A1111生成进度
a1111_generate 3090 通过txt2img生成图像
a1111_list_models 3090 列出SD检查点型号
a1111_list_loras 3090 列出A1111 LoRA型号
list_outputs any 列出目录中最近的文件
tail_log 任何 尾部服务日志
scp_pull any 通过SFTP拉取文件
zt_status 2070/3090 零层对等状态
ping_hosts all 检查SSH连接