Pin Atari version

danijar · Feb 13, 2025 · 7949c3c · 7949c3c
1 parent ec19f16
commit 7949c3c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -4,7 +4,6 @@
 
 # System
 FROM ghcr.io/nvidia/driver:56b85890-550.90.07-ubuntu22.04
-# FROM ghcr.io/nvidia/driver:dbcd761b-550.90.12-ubuntu24.04  # DMLab build fails on Ubuntu 24
 ENV DEBIAN_FRONTEND=noninteractive
 ENV TZ=America/San_Francisco
 ENV PYTHONUNBUFFERED=1
@@ -24,7 +23,7 @@ RUN pip install --upgrade pip setuptools
 
 # Envs
 RUN wget -O - https://gist.github.com/danijar/ca6ab917188d2e081a8253b3ca5c36d3/raw/install-dmlab.sh | sh
-RUN pip install ale_py autorom[accept-rom-license]
+RUN pip install ale_py==0.9.0 autorom[accept-rom-license]==0.6.1
 RUN pip install procgen_mirror
 RUN pip install crafter
 RUN pip install dm_control
@@ -38,7 +37,6 @@ RUN chown -R 1000:root /venv/lib/python3.11/site-packages/minerl
 # Requirements
 COPY requirements.txt requirements.txt
 RUN pip install -r requirements.txt
-# RUN pip install -r requirements.txt -f https://storage.googleapis.com/jax-releases/jax_nightly_releases.html
 
 # Source
 RUN mkdir /app

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,5 @@
+ale_py==0.9.0
+autorom[accept-rom-license]==0.6.1
 av
 chex
 colored_traceback