Update cupy dependency

rapidsai · May 3, 2023 · 00734e9 · 00734e9
1 parent 2dfbe83
commit 00734e9
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -15,7 +15,7 @@ dependencies:
 - cuda-profiler-api=11.8.86
 - cuda-python >=11.7.1,<12.0
 - cudatoolkit=11.8
-- cupy
+- cupy>=12.0.0
 - cxx-compiler
 - cython>=0.29,<0.30
 - dask-core==2023.3.2

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -286,6 +286,11 @@ dependencies:
     common:
       - output_types: [conda, requirements, pyproject]
         packages:
-          - cupy
           - scikit-learn
           - scipy
+      - output_types: conda
+        packages:
+          - cupy>=12.0.0
+      - output_types: pyproject
+        packages:
+          - cupy-cuda11x>=12.0.0
diff --git a/python/pylibraft/pyproject.toml b/python/pylibraft/pyproject.toml
@@ -50,7 +50,7 @@ classifiers = [
 
 [project.optional-dependencies]
 test = [
-    "cupy",
+    "cupy-cuda11x>=12.0.0",
     "pytest",
     "pytest-cov",
     "scikit-learn",