Fix jax source in setup script.

fmi-basel · Jun 24, 2022 · 2111e17 · 2111e17
1 parent 3a41d4d
commit 2111e17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup_environment.sh b/setup_environment.sh
@@ -8,7 +8,7 @@ cd $base_dir && \
 conda env create -p $(pwd)/$CONDA_ENV_NAME -f GUIFold/conda_pkgs.yml
 conda activate $(pwd)/$CONDA_ENV_NAME && \
 pip install --upgrade pip && \
-pip install jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html && \
+pip install jaxlib==0.1.69+cuda111 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html && \
 cd $(pwd)/$CONDA_ENV_NAME/lib/python3.8/site-packages && \
 patch -p0 < $base_dir/AF4GUIFold/docker/openmm.patch && \
 cd $base_dir && \