another try of jaxlib versions

adaptive-intelligent-robotics · Nov 27, 2023 · 6bba333 · 6bba333
1 parent f9837ae
commit 6bba333
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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