From 89bfdae8300ef083c34a4d576f09d56ea8c10a91 Mon Sep 17 00:00:00 2001 From: Taurean Dyer <46935140+taureandyernv@users.noreply.github.com> Date: Tue, 17 Dec 2024 15:21:40 -0800 Subject: [PATCH] Update pip-install.py --- colab/pip-install.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colab/pip-install.py b/colab/pip-install.py index 3d7c499..28dd240 100644 --- a/colab/pip-install.py +++ b/colab/pip-install.py @@ -37,7 +37,7 @@ 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}.*")