通过NVML获取GPU信息

之前有使用NVAPI获取gpu信息

参考文章

用NVAPI获取GPU信息

官方文档

但有些显卡不支持,好像专业显卡计算卡检测不到,比如T4;并且只支持windows平台

通过NVML获取GPU信息_第1张图片

NVSMI  :nvidia-smi.exe 工具 windows下查看显卡信息,包括温度、显存占用。。。

NVML 开发接口在CUDA开发包nvml文件夹中有demo,接口包含nvml.h 和 nvml.lib,但DLL在驱动目录下

通过NVML获取GPU信息_第2张图片

从NVSMI目录中可以看出 nvidia-smi.exe也是利用nvml实现的。并且最为重要的一点就是一定要使用服务器本地的nvml.dll这个动态库,因为需要和服务器的驱动要匹配,不然得不到正确的结果。

以下两个链接均有详细使用说明

NVML查询显卡信息

【Python管理GPU】pynvml工具的安装与使用

 

你可能感兴趣的:(随笔一记)