通过CUDA的Nvml库获取GPU的使用率【针对Nvidia显卡】

环境配置

• 从官网下载并默认安装CUDA SDK
• 从C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include下，包含#include "nvml.h"
• 从C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64下，包含nvml.lib
• 将 C:\Program Files\NVIDIA Corporation\NVSMI下的nvml.dll拷贝到执行目录下；

code如下

 

#include 
#include
#include
#include "nvml.h"
using namespace std;
int main()
{
    nvmlReturn_t result;
    unsigned int device_count, i;
    // First initialize NVML library
    result = nvmlInit();

    result = nvmlDeviceGetCount(&device_count);
    if (NVML_SUCCESS != result)
    {
        std::cout