From 2cb2f3dfb1d736a00a4615ed27ba1ad3c5ea0d67 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Deniz=20=C5=9Eafak?= <denizsafak98@gmail.com>
Date: Tue, 20 May 2025 04:23:22 +0300
Subject: [PATCH] Fix NVML Shared library error

---
 abogen/is_nvidia.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/abogen/is_nvidia.py b/abogen/is_nvidia.py
index 5ae457a..2777f7b 100644
--- a/abogen/is_nvidia.py
+++ b/abogen/is_nvidia.py
@@ -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())
\ No newline at end of file