From 6bba333b8b90dca15b24c72b9eb331098c4492b7 Mon Sep 17 00:00:00 2001
From: Luca Grillotti <lg4615@ic.ac.uk>
Date: Tue, 28 Nov 2023 01:11:34 +0900
Subject: [PATCH] another try of jaxlib versions

---
 dev.Dockerfile   | 2 +-
 requirements.txt | 4 ++--
 tool.Dockerfile  | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/dev.Dockerfile b/dev.Dockerfile
index 5ff639de..458599db 100644
--- a/dev.Dockerfile
+++ b/dev.Dockerfile
@@ -40,7 +40,7 @@ ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/targets/x86_64-linux/l
 
 ENV TZ=Europe/Paris
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
-RUN pip --no-cache-dir install jaxlib==0.3.15+cuda11.cudnn82 \
+RUN pip --no-cache-dir install jaxlib==0.4.16+cuda11.cudnn86 \
     -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html \
     && rm -rf /tmp/*
 
diff --git a/requirements.txt b/requirements.txt
index 16c91bc3..b66411c9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,8 +5,8 @@ dm-haiku==0.0.5
 flax==0.6.0
 gym==0.23.1
 ipython
-jax==0.3.17
-jaxlib==0.3.15
+jax==0.4.16
+jaxlib==0.4.16
 jumanji==0.1.3
 jupyter
 numpy==1.22.3
diff --git a/tool.Dockerfile b/tool.Dockerfile
index 2def88da..10b15b02 100644
--- a/tool.Dockerfile
+++ b/tool.Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.8.13-slim
+FROM python:3.9.18-slim
 
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8 PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1
 ENV PIPENV_VENV_IN_PROJECT=true PIP_NO_CACHE_DIR=false PIP_DISABLE_PIP_VERSION_CHECK=1