From 00734e91a59528ac18794e5337f5764b92cf4fe6 Mon Sep 17 00:00:00 2001 From: Vyas Ramasubramani Date: Wed, 3 May 2023 15:43:20 -0700 Subject: [PATCH] Update cupy dependency --- conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +- dependencies.yaml | 7 ++++++- python/pylibraft/pyproject.toml | 2 +- 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 8015122634..5ba52c4fad 100644 --- 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 index 5b691d1464..630ba99c59 100644 --- 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 index 3c600324ce..dc3095e2d1 100644 --- 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",