-
Notifications
You must be signed in to change notification settings - Fork 17
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
GPU installation update for SCICO #334
Comments
The significant change is in the URL for the packages? Does this still work if we keep the |
Yeah, it was something minor, I can re-do the installation and test if the upgrade argument is still fine. |
Works just fine with the --upgrade argument as long as cuda is added to the url! So it should be an easy fix to the install doc. |
Thanks for checking. I'll make the change to the docs. |
Ah ill delete the other PR, I think we did it at the same time haha. |
* Start work on overview edits * Work in progress on docs restructure * Fix residual md formatting and add missing references * Some markup improvements * Update submodule * Trivial edits * Various improvements * Various improvements * Restucture amd rename Important Classes section * Change section title * Minor change * Use more standard header level indicators * Move developers/contributors page * Rephrase some text and address #334 * Correct chapter number * Add note on limitations of non-JAX packages * Bibtex updates * Change header symbol * Move external package notes to Notes section and restructure that section
While helping Maliha, from the Imaging Lab @ Purdue, install SCICO with GPU today I found the following error in the guide:
It says the following:
To fix it all I did was update:
pip install --upgrade "jaxlib==0.3.0+cuda11.cudnn805" -f https://storage.googleapis.com/jax-releases/jax_releases.html
to:
pip install "jaxlib==0.3.0+cuda11.cudnn805" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
This was the result:
The text was updated successfully, but these errors were encountered: