Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Colab: fix pip install for nightlies #112

Merged
merged 1 commit into from
Oct 7, 2024

Conversation

jameslamb
Copy link
Contributor

@jameslamb jameslamb commented Oct 7, 2024

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:

!git clone https://github.com/jameslamb/rapidsai-csp-utils.git --branch fix/pip-install
!python rapidsai-csp-utils/colab/pip-install.py nightlies

Ran the rest of the notebook and saw it successfully install the expected versions of libraries.

!pip freeze | grep -E '.*cu.*'
image

Importing cuml failed, will open a separate issue for that and link it here. Nevermind, just a mix of versions. Everything ran successfully in a clean session!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[FEA] Add Nightlies option to pip install for colab
2 participants