From aad6294a6788cc9ca8abf323288e1ad4f45b8050 Mon Sep 17 00:00:00 2001 From: dkazanc Date: Wed, 1 May 2024 16:05:27 +0100 Subject: [PATCH] corr10 --- conda/recipe/meta.yaml | 1 - httomolibgpu/cupywrapper.py | 11 +++++++++++ pyproject.toml | 1 - 3 files changed, 11 insertions(+), 2 deletions(-) create mode 100644 httomolibgpu/cupywrapper.py diff --git a/conda/recipe/meta.yaml b/conda/recipe/meta.yaml index 5dba8804..98625e81 100644 --- 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 new file mode 100644 index 00000000..a5b858cf --- /dev/null +++ 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 \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index 2ca79416..282e66ac 100644 --- 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",