diff --git a/Dockerfile b/Dockerfile
index 810219d..de0827c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -22,7 +22,8 @@ RUN update-alternatives --install /usr/local/bin/pip pip /usr/local/bin/pip3 1
 
 RUN pip install --no-cache-dir psutil \
                 --no-cache-dir torch \
-                --no-cache-dir torchvision
+                --no-cache-dir torchvision \
+                --no-cache-dir captum
                 
 ADD serve serve
 RUN pip install ../serve/