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

Fix NVML Shared library error

Browse Source
This commit is contained in:
Deniz Şafak
2025-05-20 04:23:22 +03:00
parent 578da66f34
commit 2cb2f3dfb1
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
+5 -1
View File
@@ -1,7 +1,11 @@
import gpustat import gpustat
def check(): def check():
stats = gpustat.new_query() try:
stats = gpustat.new_query()
except Exception:
return False
for gpu in stats.gpus: for gpu in stats.gpus:
print(gpu.name) print(gpu.name)
if 'nvidia' in gpu.name.lower(): if 'nvidia' in gpu.name.lower():