From 89ba90408d0c8a8e030ad7dbf3f19b5fce78f43b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= <ageron@users.noreply.github.com>
Date: Thu, 6 Oct 2016 20:24:25 +0200
Subject: [PATCH] Add OpenAI gym

---
 Dockerfile       | 3 +++
 requirements.txt | 1 +
 2 files changed, 4 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index adbba96..c9bc6b1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,6 +2,7 @@ FROM andrewosh/binder-base
 
 USER root
 RUN apt-get update
+RUN apt-get install -y python-numpy python-dev cmake zlib1g-dev libjpeg-dev xvfb libav-tools xorg-dev python-opengl libboost-all-dev libsdl2-dev swig
 
 USER main
 
@@ -10,6 +11,8 @@ RUN conda install -c jjhelmus tensorflow=0.10.0
 RUN conda install -c conda-forge jupyter_contrib_nbextensions
 RUN jupyter contrib nbextension install --user
 RUN jupyter nbextension enable toc2/main
+RUN pip install gym gym[atari]
 
 # Python 3
 RUN conda install -n python3 -c jjhelmus tensorflow=0.10.0
+RUN /bin/bash -c "source activate python3 && pip install gym gym[atari]"
diff --git a/requirements.txt b/requirements.txt
index ef5d623..fd2c68e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+gym
 jupyter
 matplotlib
 numexpr