From b9c6efff27a0805857be823b9e882b4b9f839585 Mon Sep 17 00:00:00 2001 From: Nate Meyer Date: Tue, 27 Jun 2023 01:39:47 -0400 Subject: [PATCH] Include TensorRT version 8.5.3 --- Dockerfile | 4 ++-- requirements-tensorrt.txt | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/Dockerfile b/Dockerfile index 0115d0971..d7074bf53 100644 --- a/Dockerfile +++ b/Dockerfile @@ -263,7 +263,7 @@ WORKDIR /opt/frigate/ COPY --from=rootfs / / # Build TensorRT-specific library -FROM nvcr.io/nvidia/tensorrt:23.05-py3 AS trt-deps +FROM nvcr.io/nvidia/tensorrt:23.03-py3 AS trt-deps RUN --mount=type=bind,source=docker/support/tensorrt_detector/tensorrt_libyolo.sh,target=/tensorrt_libyolo.sh \ /tensorrt_libyolo.sh @@ -274,7 +274,7 @@ FROM frigate AS frigate-tensorrt #Disable S6 Global timeout ENV S6_CMD_WAIT_FOR_SERVICES_MAXTIME=0 -ENV TRT_VER=8.6.1 +ENV TRT_VER=8.5.3 ENV YOLO_MODELS="yolov7-tiny-416" COPY --from=trt-deps /usr/local/lib/libyolo_layer.so /usr/local/lib/libyolo_layer.so diff --git a/requirements-tensorrt.txt b/requirements-tensorrt.txt index 3b6d099a3..a6cfd9bb8 100644 --- a/requirements-tensorrt.txt +++ b/requirements-tensorrt.txt @@ -1,13 +1,13 @@ # NVidia TensorRT Support (amd64 only) --extra-index-url 'https://pypi.nvidia.com' -tensorrt == 8.6.1; platform_machine == 'x86_64' -tensorrt-libs == 8.6.1; platform_machine == 'x86_64' -tensorrt-bindings == 8.6.1; platform_machine == 'x86_64' +tensorrt == 8.5.3.*; platform_machine == 'x86_64' +# tensorrt-libs == 8.5.3.*; platform_machine == 'x86_64' +# tensorrt-bindings == 8.5.3.*; platform_machine == 'x86_64' cuda-python == 12.1; platform_machine == 'x86_64' cython == 0.29.*; platform_machine == 'x86_64' nvidia-cuda-runtime-cu12 == 12.1.*; platform_machine == 'x86_64' nvidia-cublas-cu12 == 12.1.*; platform_machine == 'x86_64' -nvidia-cudnn-cu12 == 8.9.*; platform_machine == 'x86_64' +nvidia-cudnn-cu12 == 8.8.*; platform_machine == 'x86_64' nvidia-cuda-nvrtc-cu12 == 12.1.*; platform_machine == 'x86_64' onnx==1.14.0; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64' \ No newline at end of file