GPUMonitorRGB/mainZ.py at main · edn0/GPUMonitorRGB · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
from openrgb import OpenRGBClient
from openrgb.utils import DeviceType, RGBColor
import os
import GPUtil
import time
import psutil

cli = OpenRGBClient("192.168.1.122", 6742, "My client!")

cli.show()

green = RGBColor(0, 128, 0)
yellow = RGBColor(150, 150, 0)
orange = RGBColor(255, 127, 0)
redish = RGBColor(200, 0, 0)
red = RGBColor(255, 0, 0)
pink = RGBColor(233, 39, 233)
black = RGBColor(0, 0, 0)
DRAM0 = cli.get_devices_by_type(DeviceType.DRAM)[0]
DRAM1 = cli.get_devices_by_type(DeviceType.DRAM)[1]

# Each stick has 5 invidual LEDs.
DRAM0.zones[0].resize(5)
DRAM1.zones[0].resize(5)
# Set the light mode to static.
DRAM0.set_mode(2)
DRAM1.set_mode(2)
# Turns the whole sticks off to progressively light them up later.
DRAM0.set_color(black)
DRAM1.set_color(black)

## TODO

def get_cpu_usage_pct():
    return psutil.cpu_percent(interval=0.3)

def get_cpu_load():
    CPU_load = get_cpu_usage_pct() # function from psutils
    print(CPU_load, "% CPU LOAD")

    DRAM0.zones[0].colors[4] = green
    if CPU_load >= 20:
        DRAM0.zones[0].colors[3] = yellow
    elif CPU_load < 20:
        DRAM0.zones[0].colors[3] = black
    if CPU_load >= 40:
        DRAM0.zones[0].colors[2] = orange
    elif CPU_load < 40:
        DRAM0.zones[0].colors[2] = black
    if CPU_load >= 60:
        DRAM0.zones[0].colors[1] = pink
    elif CPU_load < 60:
        DRAM0.zones[0].colors[1] = black
    if CPU_load > 80:
        DRAM0.zones[0].colors[0] = red
    elif CPU_load < 80:
        DRAM0.zones[0].colors[0] = black

def get_gpu_load():
    GPU_load = GPUs[0].load
    print(GPU_load*100, "% GPU LOAD")


def get_gpu_temps():
    GPU_temp = GPUs[0].temperature
    print(GPU_temp, "°c GPU TEMP")

    DRAM1.zones[0].colors[4] = green
    if GPU_temp >= 51:
        DRAM1.zones[0].colors[3] = yellow
    elif GPU_temp < 51:
        DRAM1.zones[0].colors[3] = black
    if GPU_temp >= 56:
        DRAM1.zones[0].colors[2] = orange
    elif GPU_temp < 56:
        DRAM1.zones[0].colors[2] = black
    if GPU_temp >= 66:
        DRAM1.zones[0].colors[1] = pink
    elif GPU_temp < 66:
        DRAM1.zones[0].colors[1] = black
    if GPU_temp > 78:
        DRAM1.zones[0].colors[0] = red
    elif GPU_temp < 78:
        DRAM1.zones[0].colors[0] = black

def get_memory():
    GPU_memory_used = GPUs[0].memoryUsed
    print(GPU_memory_used, " MEMORY USED")


while True:
    GPUs = GPUtil.getGPUs() # Looks up all GPUs.
    os.system('cls||clear') # Clears terminal for readability.
    get_cpu_load() # Using psutils measure avg CPU load over given interval and sets color to RAM stick accordingly. Prints to console as well.
    DRAM0.zones[0].show() # Apply color modification to first stick.
    get_gpu_load() # Get GPU load and print it out.
    get_memory() # Get VRAM usage and print it out.
    get_gpu_temps() # Get GPU temps using GPUtil and sets color to second RAM stick. Also printed in console.
    DRAM1.zones[0].show() # Apply color modification to second stick.
    time.sleep(0.3) # So that it doesn't refresh so often and generate useless load.