Added cupy as optional dependencies (#572)

ptycho · Aug 20, 2024 · 9b73736 · 9b73736
1 parent 49d9755
commit 9b73736
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,15 +23,15 @@ classifiers = [
 ]
 dynamic = ["version"]
 dependencies = [
-#  "python >= 3.7",
   "numpy",
   "scipy",
   "h5py",
 ]
 
 [project.optional-dependencies]
 full = ["mpi4py","matplotlib","pyzmq","pillow", "pyfftw"]
-
+cupy-cuda11x = ["mpi4py","matplotlib","pyzmq","pillow", "pyfftw", "cupy-cuda11x"]
+cupy-cuda12x = ["mpi4py","matplotlib","pyzmq","pillow", "pyfftw", "cupy-cuda12x"] 
 
 #[project.scripts]
 #"ptypy.plot" = 'scripts/ptypy.plot'