Skip to content

Commit

Permalink
[info/deps] Change CuPy detection method.
Browse files Browse the repository at this point in the history
Consequence of migration to CuPy v13.
  • Loading branch information
SepandKashani committed Mar 10, 2024
1 parent 5ae9006 commit 46b6525
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/pyxu/info/deps.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,9 @@
import cupy
import cupyx.scipy.sparse
import cupyx.scipy.sparse.linalg
except ImportError:
# CuPy is installed, but GPU drivers probably missing.

cupy.is_available() # will fail if hardware/drivers/runtime missing
except Exception:
CUPY_ENABLED = False


Expand Down

0 comments on commit 46b6525

Please sign in to comment.