diff --git a/conda/recipes/libraft_distance/meta.yaml b/conda/recipes/libraft_distance/meta.yaml index a4f2dbac94..4474629df4 100644 --- a/conda/recipes/libraft_distance/meta.yaml +++ b/conda/recipes/libraft_distance/meta.yaml @@ -45,12 +45,15 @@ requirements: - ucx-proc=*=gpu - gtest=1.10.0 - gmock + - librmm {{ minor_version }} run: + - libraft-headers {{ version }} - nccl>=2.9.9 - ucx-py {{ ucx_py_version }} - ucx-proc=*=gpu - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }} - libcusolver>=11.2.1 + - librmm {{ minor_version }} about: home: http://rapids.ai/ diff --git a/conda/recipes/libraft_headers/meta.yaml b/conda/recipes/libraft_headers/meta.yaml index a03146a7d5..aec6fa4351 100644 --- a/conda/recipes/libraft_headers/meta.yaml +++ b/conda/recipes/libraft_headers/meta.yaml @@ -44,8 +44,11 @@ requirements: - ucx-proc=*=gpu - gtest=1.10.0 - gmock + - librmm {{ minor_version}} + - libcusolver>=11.2.1 run: - nccl>=2.9.9 + - librmm {{ minor_version}} - ucx-py {{ ucx_py_version }} - ucx-proc=*=gpu - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }} diff --git a/conda/recipes/libraft_nn/meta.yaml b/conda/recipes/libraft_nn/meta.yaml index 53f44f2e97..9d6732d56b 100644 --- a/conda/recipes/libraft_nn/meta.yaml +++ b/conda/recipes/libraft_nn/meta.yaml @@ -45,11 +45,14 @@ requirements: - libfaiss 1.7.0 *_cuda - gtest=1.10.0 - gmock + - librmm {{ minor_version }} run: - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }} + - libraft-headers {{ version }} - faiss-proc=*=cuda - libfaiss 1.7.0 *_cuda - libcusolver>=11.2.1 + - librmm {{ minor_version }} about: home: http://rapids.ai/ diff --git a/conda/recipes/pyraft/meta.yaml b/conda/recipes/pyraft/meta.yaml index 4182137f85..eae9963204 100644 --- a/conda/recipes/pyraft/meta.yaml +++ b/conda/recipes/pyraft/meta.yaml @@ -33,13 +33,13 @@ requirements: - libraft-headers {{ version }} - cudatoolkit {{ cuda_version }}.* - cuda-python >=11.5,<12.0 + - nccl>=2.9.9 - ucx-py {{ ucx_py_version }} - ucx-proc=*=gpu run: - python x.x - dask-cuda {{ minor_version }} - libraft-headers {{ version }} - - cupy>=7.8.0,<10.0.0a0 - nccl>=2.9.9 - rmm {{ minor_version }} - ucx-py {{ ucx_py_version }} @@ -60,4 +60,4 @@ about: home: http://rapids.ai/ license: Apache-2.0 # license_file: LICENSE - summary: pyraft library \ No newline at end of file + summary: pyraft library