diff --git a/app/Dockerfile b/app/Dockerfile
index cbdc9e3a..370a51de 100644
--- a/app/Dockerfile
+++ b/app/Dockerfile
@@ -1,13 +1,7 @@
-FROM ubuntu:bionic-20200219 as base
+FROM tensorflow/tensorflow:1.15.5
 
-RUN apt-get update
-RUN apt-get -y install ssh
-RUN apt-get -y install python3-pip
-RUN apt-get -y install htop
-RUN apt-get -y install libpq-dev
-
-RUN apt-get update 
-RUN apt-get -y install cmake libopenmpi-dev python3-dev zlib1g-dev libgl1-mesa-dev
+RUN apt-get update -y && apt-get upgrade -y
+RUN apt-get -y install cmake libopenmpi-dev python3-dev zlib1g-dev libgl1-mesa-dev ssh python3-pip htop libpq-dev
 
 RUN pip3 install --upgrade pip
 RUN pip3 install --upgrade setuptools