Merge pull request #401 from shwina/add-cudapython

Add cuda-python
rapidsai · Dec 6, 2021 · 3b4eaa6 · 3b4eaa6
2 parents cc286ca + df14246
commit 3b4eaa6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/conda/recipes/rapids-build-env/meta.yaml b/conda/recipes/rapids-build-env/meta.yaml
@@ -55,6 +55,7 @@ requirements:
     - conda {{ conda_version }}
     - conda-build {{ conda_build_version }}
     - conda-verify {{ conda_verify_version }}
+    - cuda-python {{ cuda_python_version }}.*
     - cudatoolkit ={{ cuda_version }}.*
     - cupy {{ cupy_version }}
     - cython {{ cython_version }}

diff --git a/conda/recipes/rapids/meta.yaml b/conda/recipes/rapids/meta.yaml
@@ -35,6 +35,7 @@ requirements:
   host:
     - python
   run:
+    - cuda-python {{ cuda_python_version }}.*
     - cudatoolkit ={{ cuda_version }}.*
     - cupy {{ cupy_version }}
     - nccl {{ nccl_version }}

diff --git a/conda/recipes/versions.yaml b/conda/recipes/versions.yaml
@@ -42,6 +42,8 @@ cmake_format_version:
   - '=0.6.11'
 cmake_setuptools_version:
   - '>=0.1.3'
+cuda_python_version:
+  - '>=11.5,<12.*'
 cupy_version:
   - '>=9.5.0,<10.0.0a0'
 cython_version: