C++如何获取系统的当前GPU显存占用情况(API调用)-C++-PHP中文网

windows下用nvml获取nvidia gpu显存占用最稳定，需先装nvidia驱动（≥410.48）并链接nvidia-ml.dll；调用nvmlinit()初始化，失败则返回nvml_error_driver_not_loaded；再通过nvmldevicegethandlebyindex(0)获取句柄，调用nvmldevicegetmemoryinfo()获取used/total字段（单位字节），最后调用nvmlshutdown()。

c++如何获取系统的当前gpu显存占用情况(api调用)

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++ Code Review Master

组合式C++代码评审方案，融合静态分析、AI推理、多轮迭代评审和C++专项检查，适用于PR审查、增量代码审查、全项目评审和代码质量评分，触发词包括review cpp、cpp代码评审、C++review、代码审查。

下载

误以为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等计算卡上默认禁用，需管理员权限开启。