Bump min Rapids versions to 24.8 (NVIDIA#202)

Signed-off-by: Ayush Dattagupta <[email protected]>
sarahyurick · Aug 26, 2024 · 1415ded · 1415ded
1 parent 9a424c7
commit 1415ded
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/examples/slurm/container-entrypoint.sh b/examples/slurm/container-entrypoint.sh
@@ -57,6 +57,7 @@ if [[ $DEVICE == 'gpu' ]]; then
     --scheduler-file $SCHEDULER_FILE \
     --rmm-pool-size $RMM_WORKER_POOL_SIZE \
     --interface $INTERFACE \
+    --enable-cudf-spill \
     --rmm-async >> $WORKER_LOG 2>&1 &
 fi
 

diff --git a/setup.py b/setup.py
@@ -75,11 +75,11 @@
     ],
     extras_require={
         "cuda12x": [
-            "cudf-cu12>=24.2",
-            "dask-cudf-cu12>=24.2",
-            "cuml-cu12>=24.2",
-            "cugraph-cu12>=24.2",
-            "dask-cuda>=24.2",
+            "cudf-cu12>=24.8",
+            "dask-cudf-cu12>=24.8",
+            "cuml-cu12>=24.8",
+            "cugraph-cu12>=24.8",
+            "dask-cuda>=24.8",
             "spacy[cuda12x]>=3.6.0, <4.0.0",
         ],
     },