Decrease Docker image size

Signed-off-by: Andrej Orsula <orsula.andrej@gmail.com>

Dockerfile

@@ -8,9 +8,7 @@ SHELL ["/bin/bash", "-c"]
 ### Install dependencies
 RUN python3 -m pip install --no-cache-dir setuptools==65.5.0 pip==21.3.1 && \
     python3 -m pip install --no-cache-dir "dreamerv3 @ git+https://github.com/AndrejOrsula/dreamerv3.git" && \
-    python3 -m pip install --no-cache-dir --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html && \
-    python3 -m pip install --no-cache-dir --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121 && \
-    wget -q https://raw.githubusercontent.com/danijar/dreamerv3/main/dreamerv3/configs.yaml -O "$(pip show dreamerv3 | grep Location: | cut -d' ' -f2)/dreamerv3/configs.yaml"
+    wget -q https://raw.githubusercontent.com/AndrejOrsula/dreamerv3/main/dreamerv3/configs.yaml -O "$(pip show dreamerv3 | grep Location: | cut -d' ' -f2)/dreamerv3/configs.yaml"
 
 ### Enable GUI
 RUN apt-get update && \