Dymas/abogen
1
0
Fork 0
mirror of https://github.com/denizsafak/abogen.git synced 2026-07-18 13:40:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Better GPU detection

Browse Source
This commit is contained in:
Deniz Şafak
2025-07-06 05:17:45 +03:00
parent 6b997d2909
commit 3a87fb032f
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
abogen/is_nvidia.py
+6 -2
View File
@@ -6,11 +6,15 @@ def check():
except Exception: except Exception:
return False return False
nvidia_keywords = ['nvidia', 'rtx', 'gtx', 'quadro', 'tesla', 'titan', 'mx']
for gpu in stats.gpus: for gpu in stats.gpus:
print(gpu.name) name = gpu.name.lower()
if 'nvidia' in gpu.name.lower(): if any(keyword in name for keyword in nvidia_keywords):
return True return True
return False return False
if __name__ == "__main__": if __name__ == "__main__":
stats = gpustat.new_query()
for gpu in stats.gpus:
print(gpu.name)
print(check()) print(check())