Switch to Python 3.10

rainx0r · Jun 12, 2023 · 2d0815a · 2d0815a
1 parent 6b8ed72
commit 2d0815a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/jax-rl/Dockerfile b/jax-rl/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.11-slim-buster
+FROM python:3.10-slim-buster
 LABEL maintainer="[email protected]"
 
 RUN apt-get update && apt install -y --no-install-recommends git python3-pip

diff --git a/jax-rl/README.md b/jax-rl/README.md
@@ -4,7 +4,8 @@ A base Dockerfile containing all the required dependencies for running my RL exp
 
 Includes:
 
+-   [email protected]
 -   CUDA & cuDNN for GPU acceleration
 -   JAX and [jaxtyping](https://github.com/google/jaxtyping).
--   DeepMind's [Haiku](https://github.com/deepmind/dm-haiku), [Optax](https://github.com/deepmind/optax), [RLax](https://github.com/deepmind/rlax) and [chex](https://github.com/deepmind/chex).
+-   DeepMind's [Haiku](https://github.com/deepmind/dm-haiku), [Optax](https://github.com/deepmind/optax), [RLax](https://github.com/deepmind/rlax), [chex](https://github.com/deepmind/chex) and [Distrax](https://github.com/deepmind/chex).
 -   [Weights&Biases](https://wandb.ai) and [tensorboardX](https://github.com/lanpa/tensorboardX) for logging.