Add librmm cuda pinning.

rapidsai · Mar 28, 2023 · 7f3b1ac · 7f3b1ac
1 parent 34deae2
commit 7f3b1ac
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/conda/recipes/libraft/meta.yaml b/conda/recipes/libraft/meta.yaml
@@ -52,6 +52,9 @@ outputs:
         - {{ compiler('cuda11') }} {{ cuda_version }}
         {% else %}
         - {{ compiler('cuda') }} {{ cuda_version }}
+        - cuda-cudart ={{ cuda_version }}
+        - cuda-cudart-dev ={{ cuda_version }}
+        - cuda-cudart-static ={{ cuda_version }}
         {% endif %}
         - cmake {{ cmake_version }}
         - ninja
@@ -80,7 +83,7 @@ outputs:
         - libcusparse {{ libcusparse_host_version }}
         - libcusparse-dev {{ libcusparse_host_version }}
         {% endif %}
-        - librmm ={{ minor_version }}
+        - librmm ={{ minor_version }}*=cuda{{ cuda_major }}
       run:
         {% if cuda_major == "11" %}
         - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }}
@@ -105,7 +108,7 @@ outputs:
         - libcusparse {{ libcusparse_run_version }}
         - libcusparse-dev {{ libcusparse_run_version }}
         {% endif %}
-        - librmm ={{ minor_version }}
+        - librmm ={{ minor_version }}*=cuda{{ cuda_major }}
     about:
       home: https://rapids.ai/
       license: Apache-2.0