From bcc4da7bedce3a8d7da87f20b5eb0c1adea23beb Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Tue, 31 Dec 2024 07:40:02 -0700
Subject: [PATCH] Include necessary cudnn deps in cuda folder

---
 docker/tensorrt/Dockerfile.base | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/docker/tensorrt/Dockerfile.base b/docker/tensorrt/Dockerfile.base
index c03781cad..bf563e7d4 100644
--- a/docker/tensorrt/Dockerfile.base
+++ b/docker/tensorrt/Dockerfile.base
@@ -25,7 +25,10 @@ ENV S6_CMD_WAIT_FOR_SERVICES_MAXTIME=0
 
 #COPY --from=trt-deps /usr/local/lib/libyolo_layer.so /usr/local/lib/libyolo_layer.so
 #COPY --from=trt-deps /usr/local/src/tensorrt_demos /usr/local/src/tensorrt_demos
-COPY --from=trt-deps /usr/local/cuda-12.* /usr/local/cuda
+
+COPY --from=trt-deps /usr/lib/x86_64-linux-gnu/libcudnn* /usr/local/cuda/lib64/
+COPY --from=trt-deps /usr/lib/x86_64-linux-gnu/libnv* /usr/local/cuda/lib64/
+
 COPY docker/tensorrt/detector/rootfs/ /
 ENV YOLO_MODELS=""