diff --git a/conda/recipes/rapids-build-env/meta.yaml b/conda/recipes/rapids-build-env/meta.yaml index 9d1068de..e0f0e626 100644 --- 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 index fe2c439c..b7b13f85 100644 --- 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 index 4c8cbe1c..0ce6cd98 100644 --- 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: