diff --git a/.drone.yml b/.drone.yml
index c7bcae6..363166e 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -14,7 +14,7 @@ steps:
         from_secret: mount_command
     commands:
       - python -m pip install --upgrade pip
-      - pip install torchsummary
+      - pip install torchsummary matplotlib
       - ls ./
       - mkdir -p data
       - $mount_command