Pin RAPIDS dependencies

ryantwolf · ryantwolf · commit e927313dbc60 · 2024-05-16T12:58:25.000-07:00
Signed-off-by: Ryan Wolf &lt;rywolf@nvidia.com&gt;
diff --git a/setup.py b/setup.py
@@ -66,10 +66,10 @@
     ],
     extras_require={
         "cuda12x": [
-            "cudf-cu12>=24.2",
-            "dask-cudf-cu12>=24.2",
-            "cugraph-cu12>=24.2",
-            "dask-cuda>=24.2",
+            "cudf-cu12==24.4.*",
+            "dask-cudf-cu12==24.4.*",
+            "cugraph-cu12==24.4.*",
+            "dask-cuda==24.4.*",
             "spacy[cuda12x]>=3.6.0, <4.0.0",
         ]
     },
