Repolode
shanselman

shanselman/gputaskman

面向 Windows + NVIDIA 的终端监控工具,按进程展示显存、引擎占用及 LLM 权重与 KV 缓存构成。

Star
242
Fork
12
未关闭 Issue
1
最近提交
昨天

这是什么?

gputaskman 是一个面向 Windows 与 NVIDIA 显卡的 htop 式监控工具。任务管理器只能显示显存总量,无法说明占用来自哪里。它把 Windows GPU 计数器与 NVML 设备数据结合起来,按进程拆分显存、系统内存中可被 GPU 访问的部分、引擎活动,以及 CUDA、Vulkan、Direct3D、ggml、ONNX Runtime 等运行时。

对于本地 AI 负载,它会根据内存映射文件、启动参数和 GGUF 文件头识别候选模型,并估算权重与 KV 缓存的大小。若有 llama.cpp 的缓冲日志,或可访问 Ollama、LM Studio 的本地接口,还能补充实际分配的缓冲大小与模型运行状态。每条识别结果都标注了证据强度,工具也说明:加载了某个 DLL 或映射了某个文件,并不等于模型已在显存中运行。

界面是实时终端仪表盘,包含设备曲线图和可排序的进程表;同时支持一次性的文本与 JSON 输出,便于脚本调用。项目在 Releases 页面提供 x64 和 ARM64 的便携版,也可通过 uv 或 pip 从源码运行。

核心功能

  • 01按进程展示本地显存、系统内存中可被 GPU 访问的部分以及引擎活动。
  • 02通过内存映射文件、启动参数和 GGUF 文件头识别已加载的模型。
  • 03估算 llama.cpp 的权重与 KV 缓存大小,有缓冲日志时以日志为准。
  • 04在 Ollama、LM Studio 本地服务运行时读取模型状态。
  • 05提供实时终端界面,以及一次性的文本和 JSON 输出。

适用场景

  • 排查 Windows 机器上本地大模型或其他 AI 任务占满显存的原因。
  • 发现可能导致推理变慢的系统内存回退占用。
  • 查看 llama-server 或 Ollama 进程当前加载的模型与上下文大小。
  • 以 JSON 格式采集 GPU 使用快照,用于脚本或问题排查。

快速上手

  1. 1

    使用 WinGet 安装 Git for Windows 和 uv

    winget install --id Git.Git --exact
  2. 2

    安装 uv

    winget install --id astral-sh.uv --exact
  3. 3

    克隆仓库

    git clone https://github.com/shanselman/gputaskman.git
  4. 4

    在项目文件夹中用 uv 运行 TUI

    cd gputaskman uv run --locked gputaskman

使用前须知

  • 仅支持 Windows 和 NVIDIA;纯 CPU、纯 NPU 及云端工作负载不在范围内。
  • 需要 NVIDIA 显卡并安装当前的 Windows WDDM 驱动。
  • 项目处于早期实验阶段;模型分类和基于 GGUF 的内存细分均为估算值。
  • 多 GPU 系统的归属尚不可靠;经过验证的配置仅为单块 NVIDIA GPU。

收录于专题

标签

常见问题

gputaskman 是什么?
面向 Windows + NVIDIA 的终端监控工具,按进程展示显存、引擎占用及 LLM 权重与 KV 缓存构成。 gputaskman 是一个面向 Windows 与 NVIDIA 显卡的 htop 式监控工具。任务管理器只能显示显存总量,无法说明占用来自哪里。它把 Windows GPU 计数器与 NVML 设备数据结合起来,按进程拆分显存、系统内存中可被 GPU 访问的部分、引擎活动,以及 CUDA、Vulkan、Direct3D、ggml、ONNX Runtime 等运行时。
gputaskman 可以免费使用吗?用的什么许可证?
gputaskman 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
gputaskman 还在维护吗?
活跃维护中。gputaskman 最近一次提交是在 2026年11月6日。
gputaskman 适合用来做什么?
排查 Windows 机器上本地大模型或其他 AI 任务占满显存的原因。 发现可能导致推理变慢的系统内存回退占用。 查看 llama-server 或 Ollama 进程当前加载的模型与上下文大小。 以 JSON 格式采集 GPU 使用快照,用于脚本或问题排查。
gputaskman 怎么安装和运行?
使用 WinGet 安装 Git for Windows 和 uv: winget install --id Git.Git --exact 安装 uv: winget install --id astral-sh.uv --exact 克隆仓库: git clone https://github.com/shanselman/gputaskman.git 在项目文件夹中用 uv 运行 TUI: cd gputaskman uv run --locked gputaskman
在哪里获取 gputaskman?
源代码和安装说明在 GitHub 上:https://github.com/shanselman/gputaskman

概要由 AI 根据项目 README 生成;GitHub 数据每日更新。

相关项目