
GPU MCP Server
io.github.pmady由 GitHub 账号 pmady 发布v0.1.0更新于 Oct 24, 2026
NVIDIA GPU metrics as MCP tools — utilization, memory, temperature, power. Supports MIG.
概览
本地 MCP 服务器,把 NVIDIA GPU 指标(利用率、显存、温度、功耗、PCIe 与 NVLink 吞吐)作为工具提供给 AI 助手。
- 可以做什么
- 列出所有 GPU 及其利用率和显存信息
- 按索引或 UUID 读取单块 GPU 的详细指标,包括温度、功耗、PCIe 与 NVLink 吞吐
- 查看 PID 级别的 GPU 进程归属
- 汇总所有设备的聚合统计
- 将 MIG 实例作为独立设备报告,并共享父 GPU 的指标
- 适用场景
- 当助手需要查看所在机器的 GPU 硬件状况时适用,例如检查利用率、显存压力、温度或功耗,或把 GPU 占用归因到具体进程,而无需部署 Prometheus 或 dcgm-exporter。
- 运行要求
- 作为本地 stdio 进程运行,需要装有 NVIDIA GPU 及驱动的机器。预构建容器镜像需要 NVIDIA Container Toolkit 和 --gpus all 参数;从源码构建需要 Go 1.23+、CGO 以及 Linux 上的 NVML 头文件。未声明任何账号、API 密钥或环境变量。
安装
在 SourceWeft 中
- 打开 控制台中的 GPU MCP Server,将其添加到工作区。
- 为需要使用其工具的对话启用该服务。
Desktop only,通过 STDIO。 STDIO 服务会启动本地进程,因此需要 SourceWeft 桌面宿主。
其他 MCP 客户端
参照 仓库 中的启动说明。
README
gpu-mcp-server
[CI] [Helm] [Go Report Card] [Go Reference] [License] [DOI] [OpenSSF Scorecard] [OpenSSF Best Practices]
Note: the OpenSSF Best Practices questionnaire is in progress. Once the project entry is registered at https://www.bestpractices.dev/en, swap the static badge above for the live one:
[](https://www.bestpractices.dev/projects/<ID>)
An MCP server that exposes NVIDIA GPU metrics as tools. Any MCP-compatible AI agent (Claude, Goose, Cursor, etc.) can query real-time GPU utilization, memory, temperature, power, PCIe and NVLink throughput no Prometheus or dcgm-exporter required.
Built on the official Go MCP SDK and NVIDIA go-nvml.
Tools
All tools support MIG (Multi-Instance GPU) - MIG instances appear as separate devices with their parent GPU's shared metrics (temperature, power, PCIe).
Sample output
Each tool returns structured JSON. The examples below show the shape of the data an agent receives from a node with two NVIDIA A100 GPUs.
list_gpus:
get_gpu_metrics (with {"index": 0} or {"uuid": "GPU-aaaa-1111"}):
gpu_summary:
MIG instances add is_mig, parent_gpu, and mig_profile fields to the
get_gpu_metrics and list_gpus payloads.
Quick start
Claude Desktop
Add to claude_desktop_config.json:
Goose
Cursor
Add to .cursor/mcp.json for a project, or ~/.cursor/mcp.json for all
projects:
Windsurf
Add to ~/.codeium/windsurf/mcp_config.json:
Build
Requires Go 1.23+, CGO, and NVIDIA drivers on the target machine.
Tests use a mock collector, so they run anywhere no GPU hardware required.
Docker
Prebuilt multi-arch images (linux/amd64, linux/arm64) are published to GHCR on every release.
The host needs the NVIDIA Container Toolkit
installed for --gpus all to work. The server speaks MCP over stdio, so the
-i flag is required — don't drop it.
Pin a specific version via tag instead of :latest, e.g. ghcr.io/pmady/gpu-mcp-server:v0.1.0.
Architecture
The server runs as a local process alongside the agent. It calls NVML directly through cgo — no sidecar, no network hops, no metric pipeline to configure.
Project info
- License: Apache 2.0
- Language: Go
- AAIF project alignment: MCP
- Related: keda-gpu-scaler (GPU autoscaling for Kubernetes)
- Whitepaper: GPU-Aware Autoscaling in Cloud Native AI Infrastructure — CNCF TAG Infrastructure initiative (TOC #2188)
Citing
If you use gpu-mcp-server in your research or writing, please cite it via its DOI. Full citation metadata is in CITATION.cff.
Roadmap
See ROADMAP.md for the 12-month public roadmap.
Contributing
See CONTRIBUTING.md for how to get involved.
Contributors
Thanks to all our contributors! Add yourself via PR.
Governance
This project follows Linux Foundation Minimum Viable Governance.
Documentation
- Full documentation - hosted on Read the Docs
- ROADMAP.md - public roadmap
- GOVERNANCE.md - decision-making process
- DEPENDENCIES.md - external dependencies and licenses
- SECURITY.md - vulnerability reporting
- AGENTS.md - instructions for AI agents working on this repo
- CODE_OF_CONDUCT.md - community standards
Star History
[Star History Chart]来源:README.md,提交 58306b7
工具
0版本历史
1- v0.1.0最新Sep 16, 2026
评分
已满足适用检查项的 50%
该评分衡量此收录条目的完整程度,依据注册中心条目、端点与公开来源计算得出。它不是对服务器本身的评价,也不表示服务器安全或当前可用:可连通性是带日期的检查结果,评分也不包含安装量或用户评价。
- 近 30 天内可连通25 分
请保持端点能够响应 MCP 握手;SourceWeft 每 30 天探测一次。
- 已声明认证方式10 分
请在注册中心条目中声明凭据请求头,或保持端点能够响应探测,以便判定其认证方式。
- 已列出工具15 分
请让服务器在无凭据时响应 tools/list,或发布可在沙箱中无凭据启动的软件包。
- 描述必需15 分
- 图标10 分
- 文档15 分
- 许可证10 分