From d1b963e05d8a23f5d4a2728d43c189fc2357e094 Mon Sep 17 00:00:00 2001 From: "Michael Pilosov, PhD" Date: Tue, 13 Feb 2024 05:35:44 +0000 Subject: [PATCH] minimal requirements. --- scripts/requirements.cuml.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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