diff --git a/docker/tensorflow/Dockerfile b/docker/tensorflow/Dockerfile index 08d482736..8eda3ea91 100644 --- a/docker/tensorflow/Dockerfile +++ b/docker/tensorflow/Dockerfile @@ -59,6 +59,7 @@ RUN export uid=${HOST_UID} gid=${HOST_GID} && \ RUN git clone https://github.com/mwydmuch/ViZDoom ${HOME_DIR}/vizdoom RUN pip3 install ${HOME_DIR}/vizdoom RUN pip3 install tensorflow-gpu +RUN pip3 install matplotlib scipy scikit-image tqdm USER ${USER_NAME} WORKDIR ${HOME_DIR}