diff --git a/.drone.yml b/.drone.yml
index a4b28f3..06a0b4a 100644
--- a/.drone.yml
+++ b/.drone.yml
@@ -9,4 +9,5 @@ steps:
   - name: install
     image: nvcr.io/nvidia/pytorch:23.03-py3
     commands:
-      - pip freeze
+      - pip install torchsummary
+      - ls ./