GPU: NVML
Jump to navigation
Jump to search
Example NVML C program
#include <stdio.h>
#include "nvml.h"
int main() {
nvmlReturn_t result;
unsigned int device_count, i;
char version[80];
result = nvmlInit();
result = nvmlSystemGetDriverVersion(version,80);
printf("\n Driver version: %s \n\n", version);
result = nvmlDeviceGetCount(&device_count);
printf("Found %d device%s\n\n", device_count,
device_count != 1 ? "s" : "");
printf("Listing devices:\n");
for (i = 0; i < device_count; i++) {
nvmlDevice_t device;
char name[64];
nvmlComputeMode_t compute_mode;
result = nvmlDeviceGetHandleByIndex(i, &device);
result = nvmlDeviceGetName(device, name,
sizeof(name)/sizeof(name[0]));
printf("%d. %s \n", i, name);
}
result = nvmlShutdown();
}