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}.*")