chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

Include necessary cudnn deps in cuda folder

Browse Source
This commit is contained in:
Nicolas Mowen
2024-12-31 07:40:02 -07:00
parent 416da51302
commit bcc4da7bed
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docker/tensorrt/Dockerfile.base
View File

@@ -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/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/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/ / COPY docker/tensorrt/detector/rootfs/ /
ENV YOLO_MODELS="" ENV YOLO_MODELS=""