Fail to build docker image

[leiterenato]

If I try to build the docker image:
https://github.com/deepmind/alphafold/blob/main/docker/Dockerfile

I get the following error:

Looking in links: https://storage.googleapis.com/jax-releases/jax_releases.html
Collecting jax==0.2.14
  Downloading jax-0.2.14.tar.gz (669 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 669.2/669.2 kB 12.6 MB/s eta 0:00:00
  Preparing metadata (setup.py): started
  Preparing metadata (setup.py): finished with status 'done'
ERROR: Could not find a version that satisfies the requirement jaxlib==0.1.69+cuda111 (from versions: 0.1.23, 0.1.29, 0.1.32, 0.1.40, 0.1.41, 0.1.42, 0.1.43, 0.1.44, 0.1.46, 0.1.50, 0.1.51, 0.1.52, 0.1.55, 0.1.56, 0.1.57, 0.1.58, 0.1.59, 0.1.60, 0.1.61, 0.1.62, 0.1.63, 0.1.64, 0.1.65, 0.1.66, 0.1.67, 0.1.68, 0.1.69, 0.1.70, 0.1.71, 0.1.72, 0.1.73, 0.1.74, 0.1.75, 0.1.76, 0.3.0, 0.3.2, 0.3.5, 0.3.7, 0.3.8, 0.3.10)
ERROR: No matching distribution found for jaxlib==0.1.69+cuda111
The command '/bin/bash -c pip3 install --upgrade pip     && pip3 install -r /app/alphafold/requirements.txt     && pip3 install --upgrade       jax==0.2.14       jaxlib==0.1.69+cuda$(cut -f1,2 -d. <<< ${CUDA} | sed 's/\.//g')       -f https://storage.googleapis.com/jax-releases/jax_releases.html' returned a non-zero code: 1

Looking at https://storage.googleapis.com/jax-releases/jax_releases.html, there are no references to jaxlib 0.1.69 with cuda 11.1.

Could you please verify?

[DannySaba]

They split the cuda index out from the non-cuda index, this solution resolved the build issue for me.

#510 (comment)

[leiterenato]

Thank you.