mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 13:40:27 +02:00
Fix NVML Shared library error
This commit is contained in:
+6
-2
@@ -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())
|
||||
Reference in New Issue
Block a user