diff --git a/Dockerfile b/Dockerfile
index 3d518b5687ddb5fe7819c94ac5a714b018556a2b..3f33ae8d63c2d30f0b1aef2f2e933dcd153e9194 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,6 +10,7 @@ RUN rm requirements.txt
 RUN apt update; apt install nano
 RUN pip install jupyter
 RUN pip install netron
+RUN pip install matplotlib
 
 # Note that we expect the cloned finn directory on the host to be
 # mounted on /workspace/finn -- see run-docker.sh for an example