diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu
index 9f508bc8..1c97d04a 100644
--- a/docker/Dockerfile.ubuntu
+++ b/docker/Dockerfile.ubuntu
@@ -32,6 +32,11 @@ RUN if [ "$USE_GPU" = "True" ] ; then \
     else \
         echo 'cpu unsupport detectron2'; \
     fi
+RUN if [ "$USE_GPU" = "True" ] ; then \
+        bash /tmp/install_tiny_cuda_nn.sh; \
+    else \
+     echo 'cpu unsupport tiny_duda_nn'; \
+    fi
 
 # torchmetrics==0.11.4 for ofa
 RUN if [ "$USE_GPU" = "True" ] ; then \