Handle cuda-python in CUDA 11/12 branch

rapidsai · Jul 25, 2023 · 62928e1 · 62928e1
1 parent 7b156f3
commit 62928e1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/conda/recipes/rapids-build-env/meta.yaml b/conda/recipes/rapids-build-env/meta.yaml
@@ -46,10 +46,12 @@ requirements:
     - conda-build {{ conda_build_version }}
     - conda-verify {{ conda_verify_version }}
     - cubinlinker  # CUDA enhanced compat.
-    - cuda-python {{ cuda11_cuda_python_version }}
     - {{ pin_compatible('cuda_version', max_pin='x', min_pin='x') }}
     {% if cuda_major == "11" %}
+    - cuda-python {{ cuda11_cuda_python_version }}
     - cudatoolkit
+    {% else %}
+    - cuda-python {{ cuda12_cuda_python_version }}
     {% endif %}
     - cupy {{ cupy_version }}
     - c-compiler