diff --git a/setup.py b/setup.py
index 4f36708..3930980 100644
--- a/setup.py
+++ b/setup.py
@@ -177,6 +177,7 @@ def run(self):
     deps["torchvision"],
     deps["timm"],
     deps["scikit-image"],
+    deps["matplotlib"],
 ]
 
 setup(