From 0216e785708c6a5ef99d153737e9a402d2e4b487 Mon Sep 17 00:00:00 2001
From: Jean-Francois Pombert <jpombert@iit.edu>
Date: Thu, 23 Dec 2021 00:08:26 -0600
Subject: [PATCH] Updated jax

Changed from == to >=
---
 docker/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index bda6d3e8a..ad6e309ed 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -66,7 +66,7 @@ RUN wget -q -P /app/alphafold/alphafold/common/ \
 # Install pip packages.
 RUN pip3 install --upgrade pip \
     && pip3 install -r /app/alphafold/requirements.txt \
-    && pip3 install --upgrade jax jaxlib==0.1.69+cuda${CUDA/./} -f \
+    && pip3 install --upgrade jax jaxlib>=0.1.69+cuda${CUDA/./} -f \
       https://storage.googleapis.com/jax-releases/jax_releases.html
 
 # Apply OpenMM patch.