minimal requirements.

9 months ago · d1b963e05d
1 changed files with 3 additions and 1 deletions
--- a/scripts/requirements.cuml.sh
+++ b/scripts/requirements.cuml.sh
@ -1,3 +1,5 @@
 pip install \
    --extra-index-url=https://pypi.nvidia.com \
-    cudf-cu12==23.12.* cuml-cu12==23.12.*
+    cudf-cu12==23.12.* cuml-cu12==23.12.* \
+    dask-cudf-cu12==23.12.* pylibraft-cu12==23.12.* \
+    raft-dask-cu12==23.12.*