17 lines
356 B
C++
17 lines
356 B
C++
#include <include/nvml.h>
|
|
#include <nvidia.h>
|
|
|
|
Nvidia::Nvidia() { nvmlInit_v2(); }
|
|
Nvidia::~Nvidia() { nvmlShutdown(); }
|
|
|
|
double Nvidia::get_gpu_temperature() {
|
|
nvmlDevice_t device;
|
|
|
|
nvmlDeviceGetHandleByIndex_v2(0, &device);
|
|
|
|
unsigned int temp;
|
|
nvmlDeviceGetTemperature(device, NVML_TEMPERATURE_GPU, &temp);
|
|
|
|
return static_cast<double>(temp);
|
|
}
|