From fa088abfd6241e88eedb78fb0da0d278fd771bf0 Mon Sep 17 00:00:00 2001 From: Taurean Dyer <46935140+taureandyernv@users.noreply.github.com> Date: Tue, 17 Dec 2024 15:16:00 -0800 Subject: [PATCH] Update pip-install.py --- colab/pip-install.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/colab/pip-install.py b/colab/pip-install.py index 14cd6d8..3d7c499 100644 --- a/colab/pip-install.py +++ b/colab/pip-install.py @@ -26,12 +26,12 @@ ) gpu_name = pynvml.nvmlDeviceGetName(pynvml.nvmlDeviceGetHandleByIndex(0)) -LATEST_RAPIDS_VERSION = "24.10" -NIGHTLY_RAPIDS_VERSION = "24.12" +LATEST_RAPIDS_VERSION = "24.12" +NIGHTLY_RAPIDS_VERSION = "25.02" if(len(sys.argv)>=2): if(len(sys.argv[1])=="legacy"): - rapids_version = "24.8.*" + rapids_version = "24.10.*" print("Installing the rest of the RAPIDS " + rapids_version + " libraries") output = subprocess.Popen([f"pip install cudf-cu12=={rapids_version} cuml-cu12=={rapids_version} cugraph-cu12=={rapids_version} cuspatial-cu12=={rapids_version} cuproj-cu12=={rapids_version} cuxfilter-cu12=={rapids_version} cucim-cu12=={rapids_version} pylibraft-cu12=={rapids_version} raft-dask-cu12=={rapids_version} nx-cugraph-cu12=={rapids_version} aiohttp --extra-index-url=https://pypi.nvidia.com"], shell=True, stderr=subprocess.STDOUT, stdout=subprocess.PIPE)