Fix NVML Shared library error

This commit is contained in:
Deniz Şafak
2025-05-20 04:23:22 +03:00
parent 578da66f34
commit 2cb2f3dfb1
+6 -2
View File
@@ -1,7 +1,11 @@
import gpustat
def check():
stats = gpustat.new_query()
try:
stats = gpustat.new_query()
except Exception:
return False
for gpu in stats.gpus:
print(gpu.name)
if 'nvidia' in gpu.name.lower():
@@ -9,4 +13,4 @@ def check():
return False
if __name__ == "__main__":
print(check())
print(check())