iiiypuk/nvidia-ml
iiiypuk/nvidia-ml
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Добавлены комментарии

Browse Source
This commit is contained in:
Alexander Popov 2025-02-27 23:58:43 +03:00
parent 1ba67a44f7
commit 8867bd03ac
Signed by: iiiypuk
GPG Key ID: E47FE0AB36CD5ED6
2 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/colors.h
View File

@ -4,6 +4,7 @@
#define COLORSLIB_VERSION "1.0.0"
#ifndef NO_COLOR // With colors
// clang-format off
#define COLOR_RED(string) "\x1b[31m" string "\x1b[0m"
#define COLOR_GREEN(string) "\x1b[32m" string "\x1b[0m"
#define COLOR_YELLOW(string) "\x1b[33m" string "\x1b[0m"
@ -11,7 +12,9 @@
#define COLOR_MAGENTA(string) "\x1b[35m" string "\x1b[0m"
#define COLOR_CYAN(string) "\x1b[36m" string "\x1b[0m"
#define COLOR_RESET(string) "\x1b[0m" string "\x1b[0m"
// clang-format on
#else // Without colors
// clang-format off
#define COLOR_RED(string) string
#define COLOR_GREEN(string) string
#define COLOR_YELLOW(string) string
@ -19,6 +22,7 @@
#define COLOR_MAGENTA(string) string
#define COLOR_CYAN(string) string
#define COLOR_RESET(string) string
// clang-format on
#endif
#endif // #ifndef COLORS_H_

15
src/nv-info.c
View File

@ -7,13 +7,14 @@
int main() {
nvmlReturn_t result; // Don't used
unsigned int deviceCount;
nvmlMemory_t memory;
unsigned int power;
unsigned int temperature;
char name[NVML_DEVICE_NAME_BUFFER_SIZE];
if (nvmlInit_v2() != NVML_SUCCESS) return 1;
unsigned int deviceCount; // Хранит количество устройств (Кол-во видеокарт)
nvmlMemory_t memory; // Хранит информацию о памяти
unsigned int power; // Хранит информацию о потреблении энергии
unsigned int temperature; // Хранит информацию о температуре
char name[NVML_DEVICE_NAME_BUFFER_SIZE]; // Хранит имя устройства
if (nvmlInit_v2() != NVML_SUCCESS) return 1; // Вызывать nvmlInit_v2() и nvmlShutdown() обязательно перед всеми операциями
if (nvmlDeviceGetCount(&deviceCount) == NVML_SUCCESS) {
printf(COLOR_CYAN("Devices:") " %d\n\n", deviceCount);
@ -36,6 +37,6 @@ int main() {
}
}
if (nvmlShutdown() == NVML_SUCCESS) return 0;
if (nvmlShutdown() == NVML_SUCCESS) return 0; // Вызывать nvmlInit_v2() и nvmlShutdown() обязательно перед всеми операциями
else return 1;
}