Use __path__ instead of __file__ in load_cuda() (#77)

marella · Aug 7, 2023 · d537788 · d537788
1 parent a2a5384
commit d537788
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/ctransformers/lib.py b/ctransformers/lib.py
@@ -68,13 +68,7 @@ def load_cuda() -> bool:
         import nvidia
     except ImportError:
         return False
-    if not nvidia.__file__:
-        logger.warning(
-            "CUDA libraries might not be installed properly. "
-            "Please report at https://github.com/marella/ctransformers/issues"
-        )
-        return False
-    path = Path(nvidia.__file__).parent
+    path = Path(nvidia.__path__[0])
     system = platform.system()
     if system == "Windows":
         libs = [