install_jax

KosinskiLab · Oct 16, 2024 · 5720870 · 5720870
1 parent 3a44d3a
commit 5720870
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/github_actions.yml b/.github/workflows/github_actions.yml
@@ -50,6 +50,7 @@ jobs:
           pip install -e .
           pip install -e ColabFold --no-deps
           pip install -e alphafold --no-deps
+          pip install -U "jax[cuda12]"
           export PYTHONPATH=$PWD/AlphaLink2:$PYTHONPATH
           # install dependencies for AlphaLink backend
           # pip install torch==1.13.0+cu117 --extra-index-url https://download.pytorch.org/whl/cu117