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

最新推荐文章于 2024-07-13 18:02:15 发布

鱼儿-1226

最新推荐文章于 2024-07-13 18:02:15 发布

阅读量3.1k

点赞数

分类专栏： c++ 文章标签： c++ cuda

本文链接：https://blog.csdn.net/qq_21743659/article/details/111181490

版权

环境配置

从官网下载并默认安装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 <iostream>
#include<string>
#include<windows.h>
#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 << "Failed to query device count: " << nvmlErrorString(result);
    }
    std::cout << "Found" << device_count <<" device" << endl;

    std::cout << "Listing dev