From d08f0935291669b6d0383dd71ea1658a1257d5ea Mon Sep 17 00:00:00 2001
From: joshua saucedo <joshuansu0897@gmail.com>
Date: Fri, 29 Jun 2018 12:12:43 -0700
Subject: [PATCH] Add docker support.

---
 Dockerfile |  7 +++++++
 README.md  | 17 +++++++++++++++++
 2 files changed, 24 insertions(+)
 create mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..e1d50fe
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,7 @@
+FROM python:2
+
+WORKDIR /app
+
+COPY . .
+
+CMD [ "python", "QLearner.py" ]
\ No newline at end of file
diff --git a/README.md b/README.md
index 2ff1b00..ddb532c 100644
--- a/README.md
+++ b/README.md
@@ -29,4 +29,21 @@ Run:
 python SarsaLambdaLearner.py
 ```
 
+# Docker
+Build:
+```
+docker build -t basic-reinforcement-learning .
+```
 
+## Q-Learning
+```
+docker run -it --user $(id -u) -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix --privileged basic-reinforcement-learning python QLearner.py
+```
+## Sarsa
+```
+docker run -it --user $(id -u) -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix --privileged basic-reinforcement-learning python SarsaLearner.py
+```
+## Sarsa lambda
+```
+docker run -it --user $(id -u) -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix --privileged basic-reinforcement-learning python SarsaLambdaLearner.py
+```