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.*