Merge pull request #197 from ziembla/ziembla-mod-openaigym

OpenAI gym support in docker image

16_reinforcement_learning.ipynb

@@ -574,6 +574,15 @@
     "    plt.show()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "openai_cart_pole_rendering = False  # don't try, just use the safe way?"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 26,

docker/Dockerfile

@@ -6,12 +6,15 @@ RUN apt-get update && apt-get upgrade -y \
         build-essential \
         git \
         sudo \
+        cmake zlib1g-dev libjpeg-dev xvfb libav-tools xorg-dev libboost-all-dev libsdl2-dev swig \
     && rm -rf /var/lib/apt/lists/*
 
 RUN conda update -n base conda
 RUN conda install -y -c conda-forge \
         tensorflow \
-        jupyter_contrib_nbextensions
+        jupyter_contrib_nbextensions \
+        pyopengl
+RUN pip install "gym[atari,box2d,classic_control]"
 
 ARG username
 ARG userid