从控制台 C++ 读取 GPU 信息

Question

我想创建自己的超频监视器，我需要为其读取当前电压、时钟速度等信息。

在 C++ 中，我可以通过键入轻松地从 Nvidia-smi 获取信息，例如：

console("nvidia-smi -q -i voltage");

然后显示我：

==============NVSMI LOG==============

Timestamp                                 : Tue Dec 13 17:55:54 2022
Driver Version                            : 526.47
CUDA Version                              : 12.0

Attached GPUs                             : 1
GPU 00000000:01:00.0
    Voltage
        Graphics                          : 806.250 mV

由此我只需要电压编号，在本例中为“806.25”。

我对 <cctype> 进行了一些调查，这是我读到的内容，但我没有取得任何进展。

那么我怎样才能只将该数字导入我的 C++ 程序呢？ 我只是猜测其他命令的过程是相同的。

Answer 1

我目前没有 Nvidia GPU 来测试它（卡在 Intel 集成显卡上），所以我无法导入 cuda.h，但请随意测试它并让我知道它是否有效。

#include <iostream>
#include <chrono>
#include <cuda.h>

int main() {
  // Get the current timestamp
  auto current_time = std::chrono::system_clock::now();

  // Get the current driver version
  int driver_version;
  cudaDriverGetVersion(&driver_version);

  // Get the current CUDA version
  int cuda_version;
  cudaRuntimeGetVersion(&cuda_version);

  // Get the name of the attached GPU
  cudaDeviceProp device_properties;
  cudaGetDeviceProperties(&device_properties, 0);
  std::string gpu_name = device_properties.name;

  // Get the current voltage
  int power_usage;
  cudaDeviceGetPowerUsage(&power_usage, 0);
  int voltage = power_usage / current;

  // Output the overclocking data
  std::cout << "Timestamp: " << current_time << std::endl;
  std::cout << "Driver version: " << driver_version << std::endl;
  std::cout << "CUDA version: " << cuda_version << std::endl;
  std::cout << "Attached GPU: " << gpu_name << std::endl;
  std::cout << "Voltage: " << voltage << std::endl;

  return 0;
}

如果有效，则可以从int voltage访问您的电压。