From 993d05e8e87907750fc809bc5acd77c40524872a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20Kubov=C4=8D=C3=ADk?= <markub3327@gmail.com>
Date: Mon, 26 Feb 2024 09:38:13 +0100
Subject: [PATCH] update Dockerfile

---
 README.md         | 15 +++++++++------
 docker/Dockerfile |  4 ++--
 2 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/README.md b/README.md
index 2b73a977..05eb1e48 100644
--- a/README.md
+++ b/README.md
@@ -13,6 +13,9 @@
   * [**Reverb: A framework for experience replay**](https://arxiv.org/abs/2102.04736)
   * [**Controlling Overestimation Bias with Truncated Mixture of Continuous Distributional Quantile Critics**](https://arxiv.org/abs/2005.04269)
   * [**Acme: A Research Framework for Distributed Reinforcement Learning**](https://arxiv.org/abs/2006.00979)
+  * [**Dueling Network Architectures for Deep Reinforcement Learning**](https://arxiv.org/abs/1511.06581)
+  * [**Attention Is All You Need**](https://arxiv.org/abs/1706.03762)
+  * [**An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale**](https://arxiv.org/abs/2010.11929)
 
 ## Installation with PyPI
 
@@ -26,7 +29,7 @@
   2. Install RL-Toolkit
       ```sh
       pip3 install rl-toolkit[all]
-      ```   
+      ```
   3. Run (for **Server**)
       ```sh
       python3 -m rl_toolkit -c ./rl_toolkit/config.yaml -e MinitaurBulletEnv-v0 server
@@ -59,7 +62,7 @@
       mv ~/Downloads/bazel-3.7.2-linux-arm64 ~/bin/bazel
       chmod +x ~/bin/bazel
       export PATH=$PATH:~/bin
-      ```  
+      ```
       Clone Reverb with version that corespond with TF verion installed on NVIDIA Jetson !
       ```sh
       git clone https://github.com/deepmind/reverb
@@ -102,7 +105,7 @@
      ```sh
      -  "${PYTHON_BIN_PATH}" setup.py bdist_wheel ${PKG_NAME_FLAG} ${RELEASE_FLAG} ${TF_VERSION_FLAG} --plat manylinux2010_x86_64 > /dev/null
      +  "${PYTHON_BIN_PATH}" setup.py bdist_wheel ${PKG_NAME_FLAG} ${RELEASE_FLAG} ${TF_VERSION_FLAG}  > /dev/null
-      ```  
+      ```
       Build and install
       ```sh
       bash oss_build.sh --clean true --tf_dep_override "tensorflow~=2.9.1" --release --python "3.8"
@@ -112,12 +115,12 @@
       Cleaning
       ```sh
       cd ../
-      rm -R reverb/      
-      ```  
+      rm -R reverb/
+      ```
   3. Install RL-Toolkit
       ```sh
       pip3 install rl-toolkit
-      ```   
+      ```
   4. Run (for **Server**)
       ```sh
       python3 -m rl_toolkit -c ./rl_toolkit/config.yaml -e MinitaurBulletEnv-v0 server
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 8f0f6184..5e6b707b 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,5 +1,5 @@
 # nainstaluj Ubuntu 20.04 LTS
-FROM ubuntu:20.04
+FROM tensorflow/tensorflow:2.14.0-gpu
 
 # nastav jazyk
 ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
@@ -45,4 +45,4 @@ COPY docker/entrypoint.sh /root/
 RUN chmod +x /root/entrypoint.sh
 ENTRYPOINT ["/root/entrypoint.sh"]
 
-CMD [ "/bin/bash" ]
\ No newline at end of file
+CMD [ "/bin/bash" ]