Update Dockerfile

Dockerfile

@@ -1,11 +1,15 @@
+# Use the official image as a parent image
 FROM tensorflow/tensorflow:2.0.0-gpu-py3
 
-WORKDIR /code
+ENV MJLIB_PATH=/tf/mujoco200_linux/bin/libmujoco200.so
+ENV MJKEY_PATH=/tf/mujoco200_linux/bin/mjkey.txt
+ENV MUJOCO_GL=osmesa
+ENV LD_LIBRARY_PATH=/tf/mujoco200_linux/bin/
+ENV PYOPENGL_PLATFORM=osmesa
 
-COPY ./requirements.txt /code/requirements.txt
+# i need this for ffmpeg
+ENV PATH="/tf/.local/bin/:${PATH}"
 
-RUN apt-get update && apt-get install -y libgl1-mesa-glx libosmesa6
-RUN apt-get install python3-opengl
 RUN apt-cache search mesa
 RUN apt-get update && apt-get install -y libgl1-mesa-glx libosmesa6