corr10

DiamondLightSource · May 1, 2024 · aad6294 · aad6294
1 parent 5bec95d
commit aad6294
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/conda/recipe/meta.yaml b/conda/recipe/meta.yaml
@@ -40,7 +40,6 @@ test:
     - pytest
   imports:
     - httomolibgpu
-    - httomolibgpu.cupywrapper
     - httomolibgpu.misc
     - httomolibgpu.prep
     - httomolibgpu.recon

diff --git a/httomolibgpu/cupywrapper.py b/httomolibgpu/cupywrapper.py
@@ -0,0 +1,11 @@
+cupy_run = False
+try:
+    import cupy as cp
+    try:
+        cp.cuda.Device(0).compute_capability
+        cupy_run = True
+    except cp.cuda.runtime.CUDARuntimeError:
+        print("CuPy library is a major dependency for HTTomolibgpu, please install")
+        import numpy as cp
+except ImportError:
+    import numpy as cp
diff --git a/pyproject.toml b/pyproject.toml
@@ -5,7 +5,6 @@ build-backend = "setuptools.build_meta"
 [tool.setuptools]
 include-package-data = true
 packages = ["httomolibgpu",
-            "httomolibgpu.cupywrapper",
             "httomolibgpu.misc",
             "httomolibgpu.prep",
             "httomolibgpu.recon",