From a2ecb0ebb1ea44356b919aea33baa7231c5940d9 Mon Sep 17 00:00:00 2001 From: Ashwin Srinath Date: Fri, 3 Dec 2021 10:49:32 -0500 Subject: [PATCH 1/3] Add CUDA Python --- conda/recipes/rapids-build-env/meta.yaml | 1 + conda/recipes/rapids/meta.yaml | 1 + conda/recipes/versions.yaml | 2 ++ 3 files changed, 4 insertions(+) diff --git a/conda/recipes/rapids-build-env/meta.yaml b/conda/recipes/rapids-build-env/meta.yaml index 9d1068de..059d89d7 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..3208bdce 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: From d24d654a06ed8096c6baf8e54964f7c2080062e2 Mon Sep 17 00:00:00 2001 From: Ashwin Srinath <3190405+shwina@users.noreply.github.com> Date: Mon, 6 Dec 2021 11:22:02 -0500 Subject: [PATCH 2/3] Update conda/recipes/rapids-build-env/meta.yaml Co-authored-by: AJ Schmidt --- conda/recipes/rapids-build-env/meta.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/conda/recipes/rapids-build-env/meta.yaml b/conda/recipes/rapids-build-env/meta.yaml index 059d89d7..e0f0e626 100644 --- a/conda/recipes/rapids-build-env/meta.yaml +++ b/conda/recipes/rapids-build-env/meta.yaml @@ -55,7 +55,7 @@ requirements: - conda {{ conda_version }} - conda-build {{ conda_build_version }} - conda-verify {{ conda_verify_version }} - - cuda-python ={{ cuda_python_version }}.* + - cuda-python {{ cuda_python_version }}.* - cudatoolkit ={{ cuda_version }}.* - cupy {{ cupy_version }} - cython {{ cython_version }} From df14246d64e810f01a7ee724c9e277840b69f13e Mon Sep 17 00:00:00 2001 From: Ashwin Srinath <3190405+shwina@users.noreply.github.com> Date: Mon, 6 Dec 2021 11:22:07 -0500 Subject: [PATCH 3/3] Update conda/recipes/rapids/meta.yaml Co-authored-by: AJ Schmidt --- conda/recipes/rapids/meta.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/conda/recipes/rapids/meta.yaml b/conda/recipes/rapids/meta.yaml index 3208bdce..b7b13f85 100644 --- a/conda/recipes/rapids/meta.yaml +++ b/conda/recipes/rapids/meta.yaml @@ -35,7 +35,7 @@ requirements: host: - python run: - - cuda-python ={{ cuda_python_version }}.* + - cuda-python {{ cuda_python_version }}.* - cudatoolkit ={{ cuda_version }}.* - cupy {{ cupy_version }} - nccl {{ nccl_version }}