16 lines
396 B
Python
16 lines
396 B
Python
|
from subprocess import check_output
|
||
|
|
import json
|
||
|
|
|
||
|
|
gpu_usage = check_output(["nvidia-smi", "--query-gpu=utilization.gpu", "--format=csv,noheader,nounits"])
|
||
|
|
|
||
|
|
gpu_usage = gpu_usage.decode('utf-8').replace("\n",'')
|
||
|
|
|
||
|
|
return_json = {
|
||
|
|
"text":gpu_usage,
|
||
|
|
"alt":"GPU_PERCENT",
|
||
|
|
"tooltip":"GPU percentage",
|
||
|
|
"class":"gpu_class",
|
||
|
|
"percentage":int(gpu_usage)
|
||
|
|
}
|
||
|
|
|
||
|
|
print(json.dumps(return_json))
|