From bf6e5957109989f0b2f09d09dfecd9af89728253 Mon Sep 17 00:00:00 2001
From: ly119399 <ly119399@alibaba-inc.com>
Date: Mon, 8 Jan 2024 10:10:28 +0800
Subject: [PATCH] install tinycudann

---
 docker/Dockerfile.ubuntu | 5 +++++
 1 file changed, 5 insertions(+)

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 \