diff --git a/scripts/requirements.cuml.sh b/scripts/requirements.cuml.sh index e3bb3f8..d669fef 100644 --- 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.* \ No newline at end of file