administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 18:55:23 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Include TensorRT version 8.5.3

Browse Source
This commit is contained in:
Nate Meyer 2023-06-27 01:39:47 -04:00
parent 32170d61cf
commit b9c6efff27
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
View File

@ -263,7 +263,7 @@ WORKDIR /opt/frigate/
COPY --from=rootfs / / COPY --from=rootfs / /
# Build TensorRT-specific library # 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 \ RUN --mount=type=bind,source=docker/support/tensorrt_detector/tensorrt_libyolo.sh,target=/tensorrt_libyolo.sh \
/tensorrt_libyolo.sh /tensorrt_libyolo.sh
@ -274,7 +274,7 @@ FROM frigate AS frigate-tensorrt
#Disable S6 Global timeout #Disable S6 Global timeout
ENV S6_CMD_WAIT_FOR_SERVICES_MAXTIME=0 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" ENV YOLO_MODELS="yolov7-tiny-416"
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

8
requirements-tensorrt.txt
View File

@ -1,13 +1,13 @@
# NVidia TensorRT Support (amd64 only) # NVidia TensorRT Support (amd64 only)
--extra-index-url 'https://pypi.nvidia.com' --extra-index-url 'https://pypi.nvidia.com'
tensorrt == 8.6.1; platform_machine == 'x86_64' tensorrt == 8.5.3.*; platform_machine == 'x86_64'
tensorrt-libs == 8.6.1; platform_machine == 'x86_64' # tensorrt-libs == 8.5.3.*; platform_machine == 'x86_64'
tensorrt-bindings == 8.6.1; platform_machine == 'x86_64' # tensorrt-bindings == 8.5.3.*; platform_machine == 'x86_64'
cuda-python == 12.1; platform_machine == 'x86_64' cuda-python == 12.1; platform_machine == 'x86_64'
cython == 0.29.*; platform_machine == 'x86_64' cython == 0.29.*; platform_machine == 'x86_64'
nvidia-cuda-runtime-cu12 == 12.1.*; platform_machine == 'x86_64' nvidia-cuda-runtime-cu12 == 12.1.*; platform_machine == 'x86_64'
nvidia-cublas-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' nvidia-cuda-nvrtc-cu12 == 12.1.*; platform_machine == 'x86_64'
onnx==1.14.0; platform_machine == 'x86_64' onnx==1.14.0; platform_machine == 'x86_64'
protobuf==3.20.3; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64'