cuda libs

9 months ago · b95cb8df3d
2 changed files with 9 additions and 0 deletions
--- a/scripts/requirements.cuml.sh
+++ 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.*
--- a/scripts/requirements.rapids.sh
+++ 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.*