python - 使用 python ctypes 与 nvapi 接口(interface)(跟进演示代码)

Question

这是对我之前问题的跟进:help with python ctypes and nvapi

我发布另一个问题是因为我已经在我之前的问题中选择了一个答案并且现在问题已经改变了方向。

使用 python，我正在尝试查询 nvapi.dll 以获取我的 gpu 使用值。

这就是你在 C# 中的做法(引用如下):http://code.google.com/p/open-hardware-monitor/source/browse/trunk/Hardware/Nvidia/NVAPI.cs

因为我只需要一小部分功能，所以我真的不想换成使用另一种语言。我认为 ctypes 应该能够做到这一点......我只是不知道该怎么做。我以前从未真正使用过 Python ctypes。

为了查询 gpu 使用情况，我首先需要调用 NvAPI_Initialize 函数。还必须使用查询接口(interface)来引用函数:

from ctypes import *
nvapi = WinDLL("nvapi.dll")
nvapi_QueryInterface = nvapi.nvapi_QueryInterface

使用上面的代码，我可以访问 nvapi_QueryInterface，但我不知道如何复制这部分:

private static void GetDelegate<T>(uint id, out T newDelegate)
  where T : class
{
    IntPtr ptr = nvapi_QueryInterface(id);
    if (ptr != IntPtr.Zero)
    {
        newDelegate =
          Marshal.GetDelegateForFunctionPointer(ptr, typeof(T)) as T;
    }
    else
    {
        newDelegate = null;
    }
}

引用初始化和使用函数:

GetDelegate(0x0150E828, out NvAPI_Initialize);
GetDelegate(0x189A1FDF, out NvAPI_GPU_GetUsages);

我对此的理解可能仍然是错误的，但我花了很多时间试图解决这个问题，所以我希望有人能帮助我朝着正确的方向前进，因为我对如何进行有点迷茫.

有人可以帮助我理解如何让这一小部分代码直接在 python 中运行吗？我只需要能够在一天结束时调用 NvAPI_GPU_GetUsages 函数。

谢谢。

Answer 1

In order to query for the gpu usage I first need to call the NvAPI_Initialize function.

实际上涉及到两个初始化函数:第一个来自静态库nvapi.lib，试图动态加载nvapi.dll，定位一些函数，调用第二个来自动态链接库的初始化函数。从static库中调用NvApi_Initialize涉及的步骤大致如下:

1. 检查nvapi.dll是否已经存在加载，如果没有，加载它

2. 使用 GetProcAddress 获取指向的指针来自 DLL 的 nvapi_QueryInterface使用新获得的函数来查询带有 Id 的接口(interface)0x150E828。这将是第二个初始化函数来自动态库

3. 调用获得的初始化函数。如果功能成功(指示通过返回值0)，得到两个新函数指针通过nvapi_QueryInterface，使用IDs 0x33C7358C(我们称之为 ENTER)和 0x593E8644(EXIT)。

因此，如果您想保持纯 Python，您基本上必须使用 ctypes 复制这些步骤，这将非常简单。完成后在nvapi.dll中有一个cleanup函数，可以通过id0xD22BDD7E查询。如果您使用完该库，请记住调用此函数。

现在谈谈真正的问题:不幸的是，2010 年 12 月的 NVIDIA SDK 中没有函数 NvAPI_GPU_GetUsages。但是，NvAPI_GPU_GetDynamicPstatesInfoEx 允许您请求填充类型为 NV_GPU_DYNAMIC_PSTATES_INFO_EX 的结构数组，这反过来又会包含在给定域(即 GPU、帧缓冲区和视频引擎)的最后一秒。

基本上，如果您想检索 GPU 使用信息，您将 NV_GPU_DYNAMIC_PSTATES_INFO_EX 结构传输到 Python(请参阅 ctypes 文档中的 Structures and unions)，创建一个至少包含 3 个结构的数组并将其传递数组到 NvAPI_GPU_GetDynamicPstatesInfoEx 函数，您必须先使用 ID 0x60DED2ED 从动态链接库中查询它。

顺便说一句，这是您在初始化期间查询的两个函数发挥作用的地方 - 这是一种锁定机制。虽然不是绝对必要的(即您可以在调用期间省略它)，但我们鼓励您调用它们，即

Enter()
NvAPI_GPU_GetDynamicPstatesInfoEx(array_of_structs)
Exit()

您应该能够通过查看 NVAPI_GPU_UTILIZATION_DOMAIN_GPU 宏(或常量)找到 GPU 使用信息存储在数组中的哪个位置，不幸的是，头文件中似乎没有包含声明开发工具包。不过，您可以通过将代码结果与显示 GPU 使用信息的 NVIDIA 系统工具进行比较来轻松验证这一点。

希望对您有所帮助。