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

added jaxlib specific version #315

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

nubbthedestroyer
Copy link

In some cases, jaxlib will not install exact compatible version number. specifying version in requirements avoids startup error. I got this when running from PS on Windows 10.

The error message you're encountering, RuntimeError: jaxlib version 0.4.26 is newer than and incompatible with jax version 0.4.25. Please update your jax and/or jaxlib packages., indicates a version mismatch between jax and jaxlib. This is a common issue when the versions of these two packages are not compatible with each other. To resolve this issue, you need to ensure that both jax and jaxlib are updated to versions that are compatible.

In some cases, jaxlib will not install exact compatible version number.  specifying version in requirements avoids startup error
@pafend
Copy link

pafend commented Apr 7, 2024

For those looking to get this fixed fast, my solution was:
pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

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.

2 participants