Jaxlib installation/upgrade broken ?

Hi!

I am trying to update my current Jaxlib (1.43) to the latest version(1.47), just as you state. I have tried several things but I keep getting this warning/error and it’s not installed ( I do have internet, I can install/upgrade other packages):

Collecting jaxlib==0.1.47
  WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'NewConnectionError('<pip._vendor.urllib3.connection.VerifiedHTTPSConnection object at 0x7fa0d6033160>: Failed to establish a new connection: [Errno 101] Network is unreachable')': /jax-releases/cuda101/jaxlib-0.1.47-cp37-none-linux_x86_64.whl
  WARNING: Retrying (Retry(total=3, connect=None, read=None, redirect=None, status=None)) after connection broken by 'NewConnectionError('<pip._vendor.urllib3.connection.VerifiedHTTPSConnection object at 0x7fa0d6047c50>: Failed to establish a new connection: [Errno 101] Network is unreachable')': /jax-releases/cuda101/jaxlib-0.1.47-cp37-none-linux_x86_64.whl
  WARNING: Retrying (Retry(total=2, connect=None, read=None, redirect=None, status=None)) after connection broken by 'NewConnectionError('<pip._vendor.urllib3.connection.VerifiedHTTPSConnection object at 0x7fa0d60478d0>: Failed to establish a new connection: [Errno 101] Network is unreachable')': /jax-releases/cuda101/jaxlib-0.1.47-cp37-none-linux_x86_64.whl
  WARNING: Retrying (Retry(total=1, connect=None, read=None, redirect=None, status=None)) after connection broken by 'NewConnectionError('<pip._vendor.urllib3.connection.VerifiedHTTPSConnection object at 0x7fa0d6047940>: Failed to establish a new connection: [Errno 101] Network is unreachable')': /jax-releases/cuda101/jaxlib-0.1.47-cp37-none-linux_x86_64.whl
  WARNING: Retrying (Retry(total=0, connect=None, read=None, redirect=None, status=None)) after connection broken by 'NewConnectionError('<pip._vendor.urllib3.connection.VerifiedHTTPSConnection object at 0x7fa0d6047550>: Failed to establish a new connection: [Errno 101] Network is unreachable')': /jax-releases/cuda101/jaxlib-0.1.47-cp37-none-linux_x86_64.whl
ERROR: Could not install packages due to an EnvironmentError: HTTPSConnectionPool(host='storage.googleapis.com', port=443): Max retries exceeded with url: /jax-releases/cuda101/jaxlib-0.1.47-cp37-none-linux_x86_64.whl (Caused by NewConnectionError('<pip._vendor.urllib3.connection.VerifiedHTTPSConnection object at 0x7fa0d6047630>: Failed to establish a new connection: [Errno 101] Network is unreachable'))

My installation script:

PYTHON_VERSION=cp37  # alternatives: cp35, cp36, cp37, cp38
CUDA_VERSION=cuda101  # alternatives: cuda92, cuda100, cuda101, cuda102
PLATFORM=linux_x86_64  # alternatives: linux_x86_64
BASE_URL='https://storage.googleapis.com/jax-releases'
pip install --upgrade $BASE_URL/$CUDA_VERSION/jaxlib-0.1.47-$PYTHON_VERSION-none-$PLATFORM.whl
python -m pip install --upgrade jax  # install jax

Thanks 😃 !!

[jakevdp]

I found that running

pip install pip==20.3.3
pip install --upgrade jax jaxlib==0.1.52+cuda101 -f https://storage.googleapis.com/jax-releases/jax_releases.html

reproduces the different version in metadata error, but using older versions of pip do not result in this error. For example, this installs correctly:

pip install pip==20.2.4
pip install --upgrade jax jaxlib==0.1.52+cuda101 -f https://storage.googleapis.com/jax-releases/jax_releases.html

I suspect that the pip 20.3 dependency resolver places stricter requirements on the contents of package metadata.

[hawkinsp]

Ah! Sorry, that should have been pip install ... not pip install -e .... That said, sounds like we solved this problem. Great!