diff --git a/resources/retinal-rl.def b/resources/retinal-rl.def
index c6be54a4..4043334b 100644
--- a/resources/retinal-rl.def
+++ b/resources/retinal-rl.def
@@ -1,12 +1,17 @@
 Bootstrap: docker
-From: nvidia/cuda:11.7.1-runtime-ubuntu22.04
-#From: ubuntu:22.04
+From: ubuntu:22.04
+#nvidia/cuda:12.6.1-base-ubuntu22.04
 
 %post
 
   export TZ=Europe/Berlin
   export DEBIAN_FRONTEND=noninteractive
 
+#  # add repo for new python
+#  apt update
+#  apt install -y software-properties-common
+#  add-apt-repository ppa:deadsnakes/ppa -y
+
   # Update
   apt -qy update
 
@@ -58,7 +63,23 @@ From: nvidia/cuda:11.7.1-runtime-ubuntu22.04
 
   # retinal-rl Extra
 
-  pip3 install --no-cache-dir gymnasium==0.28.1 torch==1.13.1 vizdoom==1.2.0 torchvision==0.14.1 matplotlib opentsne pygame pycairo==1.26.1 git+https://github.com/pytorch/captum.git torchscan num2words hiyapyco omgifol git+https://github.com/alex404/sample-factory.git@fix-vtrace dpcpp-cpp-rt seaborn hydra-core networkx pylint
+  pip3 install --no-cache-dir gymnasium==0.29.1 \
+    torch==2.4.0 torchvision==0.19.0 \
+    vizdoom==1.2.4  \
+    matplotlib==3.9.1 \
+    opentsne==1.0.2 \
+    pygame==2.6.1 \
+    pycairo==1.26.1  \
+    git+https://github.com/pytorch/captum.git@fd758e025673100cb6a525d59a78893c558b825b \
+    torchinfo==1.8.0 \
+    num2words==0.5.13 \
+    omgifol==0.5.1 \
+    git+https://github.com/alex404/sample-factory.git@05465ef425530c3e2ef13dd1f9aa2d7313eb3d4a \
+    dpcpp-cpp-rt==2024.2.1 \
+    seaborn==0.13.2 \
+    hydra-core==1.3.2 \
+    networkx==3.3 \
+    pylint==3.3.1
   
   # Clean up for smaller container size
   rm acc159linux-x64.zip