Merge pull request #118 from rapidsai-community/24.12-release

24.12 updates
rapidsai-community · Dec 17, 2024 · 852e385 · 852e385
2 parents ec884a2 + 89bfdae
commit 852e385
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/colab/pip-install.py b/colab/pip-install.py
@@ -26,18 +26,18 @@
   )
 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)
   elif(sys.argv[1] == "latest"):
     print(f"Installing RAPIDS Stable {LATEST_RAPIDS_VERSION}.*")
-    output = subprocess.Popen([f"pip install cudf-cu12=={LATEST_RAPIDS_VERSION}.* cuml-cu12=={LATEST_RAPIDS_VERSION}.* cugraph-cu12=={LATEST_RAPIDS_VERSION}.* cuspatial-cu12=={LATEST_RAPIDS_VERSION} cuproj-cu12=={LATEST_RAPIDS_VERSION} cuxfilter-cu12=={LATEST_RAPIDS_VERSION} cucim-cu12=={LATEST_RAPIDS_VERSION} pylibraft-cu12=={LATEST_RAPIDS_VERSION} raft-dask-cu12=={LATEST_RAPIDS_VERSION}.* nx-cugraph-cu12=={LATEST_RAPIDS_VERSION}.* aiohttp --extra-index-url=https://pypi.nvidia.com"], shell=True, stderr=subprocess.STDOUT, 
+    output = subprocess.Popen([f"pip install cudf-cu12=={LATEST_RAPIDS_VERSION}.* cuml-cu12=={LATEST_RAPIDS_VERSION}.* cugraph-cu12=={LATEST_RAPIDS_VERSION}.* cuspatial-cu12=={LATEST_RAPIDS_VERSION} cuproj-cu12=={LATEST_RAPIDS_VERSION} cuxfilter-cu12=={LATEST_RAPIDS_VERSION} cucim-cu12=={LATEST_RAPIDS_VERSION} pylibraft-cu12=={LATEST_RAPIDS_VERSION} raft-dask-cu12=={LATEST_RAPIDS_VERSION}.* nx-cugraph-cu12=={LATEST_RAPIDS_VERSION}.* aiohttp --extra-index-url=https://pypi.nvidia.com"], shell=True, stderr=subprocess.STDOUT,
       stdout=subprocess.PIPE)
   elif(sys.argv[1] == "nightlies"):
     print(f"Installing RAPIDS {NIGHTLY_RAPIDS_VERSION}.*")