diff --git a/Dockerfile b/Dockerfile
index e929ccc2c4b8c4b6483bd26a62c83938ba4040d5..975783faa5f11a10119105ab897b17d5065cad92 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,5 @@
-FROM tensorflow/tensorflow:latest-gpu-jupyter   
+FROM tensorflow/tensorflow:latest-py3
 MAINTAINER Spencer Chen
 RUN pip install --upgrade pip
 RUN pip install -q tensorflow==2.0.0-beta1
+RUN pip install matplotlib