diff --git a/Dockerfile b/Dockerfile
index a5799f0..292cf26 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -21,6 +21,7 @@ RUN python3 -m pip install pandas
 RUN python3 -m pip install wheel --no-deps -U
 RUN python3 -m pip install torch
 RUN python3 -m pip install sklearn
+RUN python3 -m pip install matplotlib
 RUN python3 -m pip freeze
 
 ENV PATH="/root/.local/bin:${PATH}"