From 6b7ef8812de9dfcd2278063a72ace386ce99a648 Mon Sep 17 00:00:00 2001 From: Nate Meyer Date: Sat, 17 Jun 2023 23:57:08 -0400 Subject: [PATCH] Update to latest tensorrt (8.6.1) release --- docker/tensorrt_models.sh | 2 +- docs/docs/configuration/detectors.md | 2 +- requirements-tensorrt.txt | 12 ++++++------ 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/docker/tensorrt_models.sh b/docker/tensorrt_models.sh index 957e817d6..5c09a79fe 100755 --- a/docker/tensorrt_models.sh +++ b/docker/tensorrt_models.sh @@ -11,7 +11,7 @@ echo "Generating the following TRT Models: ${YOLO_MODELS:="yolov4-tiny-288,yolov mkdir -p ${OUTPUT_FOLDER} # Install packages -pip install --upgrade pip && pip install onnx==1.9.0 protobuf==3.20.3 +pip install --upgrade pip && pip install onnx==1.15.0 protobuf==3.20.3 # Clone tensorrt_demos repo git clone --depth 1 https://github.com/yeahme49/tensorrt_demos.git /tensorrt_demos diff --git a/docs/docs/configuration/detectors.md b/docs/docs/configuration/detectors.md index 84507321c..948cb9b9c 100644 --- a/docs/docs/configuration/detectors.md +++ b/docs/docs/configuration/detectors.md @@ -200,7 +200,7 @@ To generate model files, create a new folder to save the models, download the sc mkdir trt-models wget https://github.com/blakeblackshear/frigate/raw/master/docker/tensorrt_models.sh chmod +x tensorrt_models.sh -docker run --gpus=all --rm -it -v `pwd`/trt-models:/tensorrt_models -v `pwd`/tensorrt_models.sh:/tensorrt_models.sh nvcr.io/nvidia/tensorrt:22.07-py3 /tensorrt_models.sh +docker run --gpus=all --rm -it -v `pwd`/trt-models:/tensorrt_models -v `pwd`/tensorrt_models.sh:/tensorrt_models.sh nvcr.io/nvidia/tensorrt:23.05-py3 /tensorrt_models.sh ``` The `trt-models` folder can then be mapped into your Frigate container as `trt-models` and the models referenced from the config. diff --git a/requirements-tensorrt.txt b/requirements-tensorrt.txt index 90517babd..2ce288fdc 100644 --- a/requirements-tensorrt.txt +++ b/requirements-tensorrt.txt @@ -1,9 +1,9 @@ # NVidia TensorRT Support (amd64 only) nvidia-pyindex; platform_machine == 'x86_64' -nvidia-tensorrt == 8.4.1.5; platform_machine == 'x86_64' -cuda-python == 11.7; platform_machine == 'x86_64' +tensorrt == 8.6.1; platform_machine == 'x86_64' +cuda-python == 12.1; platform_machine == 'x86_64' cython == 0.29.*; platform_machine == 'x86_64' -nvidia-cuda-runtime-cu11 == 11.7.*; platform_machine == 'x86_64' -nvidia-cublas-cu11 == 11.11.*; platform_machine == 'x86_64' -nvidia-cudnn-cu11 == 8.7.*; platform_machine == 'x86_64' -nvidia-cuda-nvrtc-cu11 == 11.7.*; platform_machine == 'x86_64' \ No newline at end of file +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-cuda-nvrtc-cu12 == 12.1.*; platform_machine == 'x86_64' \ No newline at end of file