masq/i3pystatus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

utils/gpu: fix nvidia-smi parsing for new nvidia-smi that uses (N/A) when unsupported (#776)

Browse Source
This commit is contained in:
Mathis Felardos 2020-04-02 19:28:21 +02:00 committed by GitHub
parent db14a88fc9
commit 25f43621ed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
i3pystatus/utils/gpu.py
View File

@ -1,11 +1,21 @@
import subprocess import subprocess
from collections import namedtuple from collections import namedtuple
from typing import Optional
GPUUsageInfo = namedtuple('GPUUsageInfo', ['total_mem', 'avail_mem', 'used_mem', GPUUsageInfo = namedtuple('GPUUsageInfo', ['total_mem', 'avail_mem', 'used_mem',
'temp', 'percent_fan', 'temp', 'percent_fan',
'usage_gpu', 'usage_mem']) 'usage_gpu', 'usage_mem'])
def _convert_nvidia_smi_value(value) -> Optional[int]:
value = value.lower()
# If value contains 'not' or 'N/A' - it is not supported for this GPU
# (in fact, for now nvidia-smi returns '[Not Supported]' or '[N/A]' depending of its version)
if "not" in value or "n/a" in value:
return None
return int(value)
def query_nvidia_smi(gpu_number) -> GPUUsageInfo: def query_nvidia_smi(gpu_number) -> GPUUsageInfo:
""" """
:return: :return:
@ -37,7 +47,6 @@ def query_nvidia_smi(gpu_number) -> GPUUsageInfo:
output = output.decode('utf-8').split("\n")[gpu_number].strip() output = output.decode('utf-8').split("\n")[gpu_number].strip()
values = output.split(", ") values = output.split(", ")
# If value contains 'not' - it is not supported for this GPU (in fact, for now nvidia-smi returns '[Not Supported]') values = [_convert_nvidia_smi_value(value) for value in values]
values = [None if ("not" in value.lower()) else int(value) for value in values]
return GPUUsageInfo(*values) return GPUUsageInfo(*values)