Fix the bug that libcusolver.so.11 exist but libcusolver.so does not (#…

…317)
pyscf · Jan 24, 2025 · 1593f2a · 1593f2a
1 parent fb5326d
commit 1593f2a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gpu4pyscf/lib/cusolver.py b/gpu4pyscf/lib/cusolver.py
@@ -16,11 +16,13 @@
 import numpy as np
 import cupy
 import ctypes
+from ctypes.util import find_library
 from cupy_backends.cuda.libs import cusolver
 from cupy_backends.cuda.libs import cublas
 from cupy.cuda import device
 
-libcusolver = ctypes.CDLL('libcusolver.so')
+libcusolver = find_library('cusolver')
+libcusolver = ctypes.CDLL(libcusolver)
 
 CUSOLVER_EIG_TYPE_1 = 1
 CUSOLVER_EIG_TYPE_2 = 2