Colab: fix pip install for nightlies #112
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Fixes #104
While testing the RAPIDS deployment docs (rapidsai/deployment#434), I tried to follow the Colab instructions for installing RAPIDS libraries with
pip
.Found that the script that does that is broken... it has a hard-coded upper bound pin in the set of requirements generated when you ask for nightlies to be installed, resulting in impossible requirements like
cudf>=24.10.0a0,<=24.8
.This fixes that, and puts the "latest" and "nightly" versions into module-level constants to make such mistakes less likely in the future.
Notes for Reviewers
How I tested this
Opened the RAPIDS pip colab notebook at https://nvda.ws/3XEO6hK.
Pointed the install lines at my branch here:
Ran the rest of the notebook and saw it successfully install the expected versions of libraries.
ImportingNevermind, just a mix of versions. Everything ran successfully in a clean session!cuml
failed, will open a separate issue for that and link it here.