From 6d0a109e46e46a60f83c352b81d88b2e0f01abc6 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Wed, 31 Jan 2024 13:48:23 -0600
Subject: [PATCH] Pin cuda-python=12.0.* for CUDA 12 builds.

---
 conda/recipes/rapids/meta.yaml | 3 +++
 conda/recipes/versions.yaml    | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/conda/recipes/rapids/meta.yaml b/conda/recipes/rapids/meta.yaml
index baa8534e..05b519aa 100644
--- 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
index 5da651c9..b77e3361 100644
--- 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: