From b95cb8df3da0af4a6210879c60e53b190f30588c Mon Sep 17 00:00:00 2001 From: "Michael Pilosov, PhD" Date: Tue, 13 Feb 2024 05:28:18 +0000 Subject: [PATCH] cuda libs --- scripts/requirements.cuml.sh | 3 +++ scripts/requirements.rapids.sh | 6 ++++++ 2 files changed, 9 insertions(+) create mode 100644 scripts/requirements.cuml.sh create mode 100644 scripts/requirements.rapids.sh diff --git a/scripts/requirements.cuml.sh b/scripts/requirements.cuml.sh new file mode 100644 index 0000000..e3bb3f8 --- /dev/null +++ b/scripts/requirements.cuml.sh @@ -0,0 +1,3 @@ +pip install \ + --extra-index-url=https://pypi.nvidia.com \ + cudf-cu12==23.12.* cuml-cu12==23.12.* diff --git a/scripts/requirements.rapids.sh b/scripts/requirements.rapids.sh new file mode 100644 index 0000000..55f8861 --- /dev/null +++ b/scripts/requirements.rapids.sh @@ -0,0 +1,6 @@ +pip install \ + --extra-index-url=https://pypi.nvidia.com \ + cudf-cu12==23.12.* dask-cudf-cu12==23.12.* cuml-cu12==23.12.* \ + cugraph-cu12==23.12.* cuspatial-cu12==23.12.* cuproj-cu12==23.12.* \ + cuxfilter-cu12==23.12.* cucim-cu12==23.12.* pylibraft-cu12==23.12.* \ + raft-dask-cu12==23.12.*