Windows下用NVML获取NVIDIA GPU显存占用最稳定，需先装NVIDIA驱动（≥410.48）并链接nvidia-ml.dll；调用nvmlInit()初始化，失败则返回NVML_ERROR_DRIVER_NOT_LOADED；再通过nvmlDeviceGetHandleByIndex(0)获取句柄，调用nvmlDeviceGetMemoryInfo()获取used/total字段（单位字节），最后调用nvmlShutdown()。
Windows下用NVML获取NVIDIA GPU显存占用
直接调用NVIDIA官方库NVML是最稳定的方式，但必须满足两个前提：系统装了NVIDIA驱动（≥410.48），且你的程序链接了nvidia-ml.dll（或静态链接libnvidia-ml.a）。不是所有机器都装了NVIDIA驱动，所以别假设nvmlInit()一定成功——它失败时返回NVML_ERROR_DRIVER_NOT_LOADED，得先检查。
关键步骤：
- 调用
nvmlInit()初始化，失败就跳过GPU监控 - 用
nvmlDeviceGetHandleByIndex(0)获取第一块GPU句柄（索引从0开始） - 调用
nvmlDeviceGetMemoryInfo()，返回结构体含used和total字段（单位字节） - 记得最后调用
nvmlShutdown()，否则多次运行可能卡住
示例片段（错误检查已简化）：
nvmlReturn_t ret = nvmlInit();
if (NVML_SUCCESS != ret) return; // 驱动未加载或库找不到
nvmlDevice_t device;
ret = nvmlDeviceGetHandleByIndex(0, &device);
if (NVML_SUCCESS == ret) {
nvmlMemory_t mem;
ret = nvmlDeviceGetMemoryInfo(device, &mem);
if (NVML_SUCCESS == ret) {
printf("GPU memory: %zu / %zu MB\n", mem.used / 1024 / 1024, mem.total / 1024 / 1024);
}
}
Windows上用WMI查AMD/Intel GPU显存（不准确但可用）
WMI没有标准GPU显存接口，Win32_VideoController类只提供AdapterRAM（显卡总显存），不是实时占用。想“凑合看”，只能读这个值，但它永远不变，和当前使用量无关。真正想监控AMD GPU，得走ADL SDK（已停更）或AMD GPU Services (AGS)；Intel则依赖Intel Graphics API或oneAPI Level Zero——这些都要额外集成、授权、且版本碎片严重。
立即学习“C++免费学习笔记（深入）”；
常见误区：
- 误以为
Win32_VideoController.AdapterRAM能反映实时占用 → 实际是固定硬件参数 - 用
GetPerformanceInfo查系统内存，误当GPU内存 → 它只管主存 - 调用
dxgi.dll枚举适配器后直接读显存 → DXGI本身不暴露占用数据，除非你Hook D3D资源创建/释放
Linux下用sysfs或nvidia-smi间接获取
Linux没统一API，NVIDIA用户最实用的是解析nvidia-smi -q -d MEMORY输出，或读/proc/driver/nvidia/gpus/0000:01:00.0/information（路径含PCI地址）和/proc/driver/nvidia/gpus/0000:01:00.0/memory（需root权限）。注意：/proc/driver/nvidia目录只在NVIDIA驱动加载后存在，且不同内核版本路径可能变化。
安全做法：
- 优先尝试执行
nvidia-smi命令并解析stdout（无需root，兼容性好） - 避免硬编码PCI路径，用
ls /proc/driver/nvidia/gpus/动态发现设备 - 读
/sys/class/drm/card0/device/mem_info_vram_used仅适用于部分AMD开源驱动（amdgpu），且字段名在不同内核版本中可能叫mem_info_vram_total或vram_*
跨平台方案为什么难做？
根本原因在于GPU厂商不提供通用标准接口。NVIDIA有NVML，AMD有ROCm SMI（Linux only）和AGS（Windows），Intel用Level Zero，而开源Mesa/Vulkan层又只暴露设备能力，不暴露运行时占用。即使你封装一层抽象，也得在编译时或运行时动态加载对应厂商的SDK，还得处理许可证（NVML允许免费分发，但AGS要求注册开发者账号）。
实际建议：
- 明确目标GPU型号和部署环境，不要写“通用”代码
- Windows + NVIDIA场景，老老实实链NVML，别折腾WMI或DXGI
- Linux服务端监控，用
nvidia-smi --query-gpu=memory.used,memory.total --format=csv,noheader,nounits最省事 - 如果程序要跑在无GPU或集显机器上，显存监控模块必须可选、可关闭、且失败时不崩溃
最常被忽略的一点：GPU显存占用不是进程级的，而是整个GPU设备共享的，nvmlDeviceGetMemoryInfo()返回的是全局已用显存，不是你当前进程独占的部分——想定位到具体进程，得用nvmlDeviceGetComputeRunningProcesses()再逐个累加，而且它在Tesla/T4等计算卡上默认禁用，需管理员权限开启。
