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
|
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():
|
||||||
@@ -9,4 +13,4 @@ def check():
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
print(check())
|
print(check())
|
||||||
Reference in New Issue
Block a user