Pin cuda-python=12.0.* for CUDA 12 builds.

rapidsai · Jan 31, 2024 · 6d0a109 · 6d0a109
1 parent 9897d67
commit 6d0a109
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/conda/recipes/rapids/meta.yaml b/conda/recipes/rapids/meta.yaml
@@ -29,6 +29,9 @@ requirements:
     - cuda-version ={{ cuda_version }}
   run:
     - {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
+    {% if cuda_major != "11" %}
+    - cuda-python {{ cuda12_cuda_python_version }}
+    {% endif %}
     - cupy {{ cupy_version }}
     - nccl {{ nccl_version }}
     - networkx {{ networkx_version }}

diff --git a/conda/recipes/versions.yaml b/conda/recipes/versions.yaml
@@ -4,6 +4,8 @@
 xgboost_version:
   - '=2.0.3'
 
+cuda12_cuda_python_version:
+  - '=12.0.*'
 cupy_version:
   - '>=12.0.0'
 nccl_version: