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=""