讲师中心 微信公众号
AI工具推荐 视频效率加速

C++如何读取GPU显存信息?(CUDA/NVAPI调用)

轻宇大大_5131

轻宇大大_5131

发布时间:2026-02-24 16:25:14

|

923人浏览过

|

来源于php中文网

原创

NVML是C++程序中实时读取GPU显存最稳定的选择,需正确初始化、处理权限与驱动依赖,避免使用nvidia-smi解析或CUDA Runtime API等不适用方案。

c++如何读取gpu显存信息?(cuda/nvapi调用)

用 nvidia-smi 快速查显存用量,但不是编程接口

想在 C++ 程序里实时读 GPU 显存,别直接 parse nvidia-smi 输出——它启动慢、格式不稳定、没权限时会失败,只适合调试或脚本临时看一眼。真要集成进程序,得走正式 API。

用 NVML(libnvidia-ml.so)读显存,最稳

NVML 是 NVIDIA 官方提供的轻量级管理库,不依赖 CUDA 上下文,进程无需 GPU 计算能力也能调,兼容性好(从 Tesla 到 RTX 都行),且线程安全。

常见错误现象:nvmlInit() 返回 NVML_ERROR_LIBRARY_NOT_FOUND —— 没装驱动或没把 libnvidia-ml.so 所在路径加进 LD_LIBRARY_PATH;返回 NVML_ERROR_NO_PERMISSION —— 普通用户默认不能访问设备状态,需加 sudo 或配 udev 规则。

  • 初始化后必须调 nvmlDeviceGetHandleByIndex(0, &device) 获取设备句柄,索引从 0 开始,不是 PCI ID
  • 显存总量和已用值用 nvmlDeviceGetMemoryInfo(device, &info) 拿,info.used 和 info.total 单位是字节
  • 别漏掉 nvmlShutdown(),否则多次运行可能卡住设备句柄

CUDA Runtime API 能读,但有隐含前提

cudaMemGetInfo() 看起来最顺手,但它只返回当前 CUDA 上下文绑定的 GPU 的显存——如果没调过 cudaSetDevice() 或没触发过任何 kernel 启动,上下文可能根本没创建,这时调用会返回 cudaErrorInvalidValue。

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载

立即学习“C++免费学习笔记(深入)”;

使用场景:你本来就在写 CUDA 程序,顺便查一下自己用的那块卡,没问题;纯 CPU 程序想查 GPU,别选它。

  • 必须先确保至少一个 device 被 cudaSetDevice() 指定过,否则 cudaMemGetInfo() 不可靠
  • 返回的 free 和 total 是当前 context 视角下的值,受 CUDA 内存池(如 cudaMallocAsync)影响,不一定等于 NVML 的物理值
  • Windows 下若链接 cudart 静态库,要注意运行时 DLL 版本匹配,否则 cudaGetLastError() 可能报 unknown error

NVAPI 在 Windows 上可用,但限制多

NVAPI 主要面向游戏/显示控制,读显存只是副业。它需要显卡驱动带 NVAPI 支持(新版基本都带),但默认禁用——得手动调 NvAPI_Initialize(),而且只支持单 GPU 查询(多卡时行为未定义)。

容易踩的坑:NvAPI_GPU_GetMemoryInfo() 返回的 dedicatedVideoMemory 是硬编码的显卡标称值,不是实时用量;真正可用的是 availableDedicatedVideoMemory,但这个字段在某些驱动版本里恒为 0。

  • 必须用 NvAPI_Initialize() 初始化,失败就别往下走了,不能靠 LoadLibrary("nvapi.dll") 就完事
  • 调用前得用 NvAPI_SYS_GetDisplayDriverVersion() 确认驱动版本 ≥ 381.65,老驱动不支持内存查询
  • 返回结构体里的单位是 KB,不是字节,别直接当 size_t 用
NVML 是目前最靠谱的选择,但它的“靠谱”建立在你愿意处理驱动依赖和权限配置上——这点常被忽略,结果程序在测试机跑得好好的,一上线就 NVML_ERROR_DRIVER_NOT_LOADED。

热门AI工具

更多
立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

相关专题

更多
scripterror怎么解决
scripterror怎么解决

scripterror的解决办法有检查语法、文件路径、检查网络连接、浏览器兼容性、使用try-catch语句、使用开发者工具进行调试、更新浏览器和JavaScript库或寻求专业帮助等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

989

2023.10.18

500error怎么解决
500error怎么解决

500error的解决办法有检查服务器日志、检查代码、检查服务器配置、更新软件版本、重新启动服务、调试代码和寻求帮助等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2940

2023.10.25

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

4734

2025.06.09

golang结构体方法
golang结构体方法

本专题整合了golang结构体相关内容,请阅读专题下面的文章了解更多。

4871

2025.07.04

硬盘接口类型介绍
硬盘接口类型介绍

硬盘接口类型有IDE、SATA、SCSI、Fibre Channel、USB、eSATA、mSATA、PCIe等等。详细介绍:1、IDE接口是一种并行接口,主要用于连接硬盘和光驱等设备,它主要有两种类型:ATA和ATAPI,IDE接口已经逐渐被SATA接口;2、SATA接口是一种串行接口,相较于IDE接口,它具有更高的传输速度、更低的功耗和更小的体积;3、SCSI接口等等。

3148

2023.10.19

PHP接口编写教程
PHP接口编写教程

本专题整合了PHP接口编写教程,阅读专题下面的文章了解更多详细内容。

4909

2025.10.17

php8.4实现接口限流的教程
php8.4实现接口限流的教程

PHP8.4本身不内置限流功能,需借助Redis(令牌桶)或Swoole(漏桶)实现;文件锁因I/O瓶颈、无跨机共享、秒级精度等缺陷不适用高并发场景。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

3789

2025.12.29

java接口相关教程
java接口相关教程

本专题整合了java接口相关内容,阅读专题下面的文章了解更多详细内容。

446

2026.01.19

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

80

2026.10.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn