Pin RAPIDS dependencies

Signed-off-by: Ryan Wolf <[email protected]>
NVIDIA · May 16, 2024 · e927313 · e927313
1 parent 82c0f4d
commit e927313
Showing 1 changed file with 4 additions and 4 deletions.
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",
         ]
     },