Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Installation instructions not working for CUDA 11.4

See original GitHub issue

This is with python 3.9.6 and pip 21.2.4.

On the README it says

pip install --upgrade

Easy enough. Then:

pip install --upgrade "jax[cuda111]" -f https://storage.googleapis.com/jax-release.html

This ran into an error because of the jaxlib dependency:

Looking in links: https://storage.googleapis.com/jax-release.html
Collecting jax[cuda111]
  Using cached jax-0.2.19-py3-none-any.whl
Requirement already satisfied: opt-einsum in /home/eben/anaconda3/envs/refgrads/lib/python3.9/site-packages (from jax[cuda111]) (3.3.0)
Requirement already satisfied: numpy>=1.18 in /home/eben/anaconda3/envs/refgrads/lib/python3.9/site-packages (from jax[cuda111]) (1.20.3)
Requirement already satisfied: absl-py in /home/eben/anaconda3/envs/refgrads/lib/python3.9/site-packages (from jax[cuda111]) (0.13.0)
  Using cached jax-0.2.18.tar.gz (708 kB)
  Using cached jax-0.2.17.tar.gz (693 kB)
  Using cached jax-0.2.16.tar.gz (680 kB)
  Using cached jax-0.2.15.tar.gz (681 kB)
ERROR: Packages installed from PyPI cannot depend on packages which are not also hosted on PyPI.
jax depends on jaxlib@ https://storage.googleapis.com/jax-releases/cuda111/jaxlib-0.1.67+cuda111-cp39-none-manylinux2010_x86_64.whl

I tried pip install jaxlib and then tried re-installing with the same command. That did not work, so I uninstalled jaxlib and then ran pip install jaxlib==0.1.67, but trying to installing jax itself still yields the above error message.

Issue Analytics

State:
Created 2 years ago
Comments:10

Top GitHub Comments

1reaction

jakevdpcommented, Aug 28, 2021

Can you try pinning the JAX version? i.e.

pip install "jax[cuda111]==0.2.19" -f https://storage.googleapis.com/jax-releases/jax_releases.html

Also, double check your releases URL; the URL is https://storage.googleapis.com/jax-releases/jax_releases.html, above you typed https://storage.googleapis.com/jax-release.html. Using the wrong URL would make pip unable to find the jaxlib wheels, and it would keep trying older jax versions until it eventually hits this error on 0.2.15.

If the URL is correct, then I suspect it has to do with pip’s new(ish) dependency resolver trying to satisfy version requirements for other packages you have installed.

0reactions

jakevdpcommented, Aug 28, 2021

Great, glad it’s working! Sorry I didn’t notice that URL typo earlier.