diff --git a/docker/torch1.9.0-cuda10.2.dockerfile b/docker/torch1.9.0-cuda10.2.dockerfile
index ab930647c..401ac0b44 100644
--- a/docker/torch1.9.0-cuda10.2.dockerfile
+++ b/docker/torch1.9.0-cuda10.2.dockerfile
@@ -72,7 +72,9 @@ RUN pip install --no-cache-dir \
       typeguard \
       numpy \
       pytest \
-      graphviz
+      graphviz \
+      tqdm
+
 
 RUN git clone https://github.com/k2-fsa/icefall /workspace/icefall && \
     cd /workspace/icefall && \