From e1b075a11d67112384ed7f3b4fc84ade61b6fa5d Mon Sep 17 00:00:00 2001 From: Nate Meyer Date: Thu, 29 Dec 2022 22:58:26 -0500 Subject: [PATCH] Update some CUDA libraries to clean up version warning --- frigate/detectors/plugins/tensorrt.py | 1 - requirements-tensorrt.txt | 9 +++------ 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 8122d1833..71ce9b048 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -315,5 +315,4 @@ class TensorRtDetector(DetectionApi): detections, np.zeros((append_cnt, 6), np.float32), axis=0 ) - # self.fps.update() return detections diff --git a/requirements-tensorrt.txt b/requirements-tensorrt.txt index c109e33d1..1a53892c0 100644 --- a/requirements-tensorrt.txt +++ b/requirements-tensorrt.txt @@ -3,9 +3,6 @@ nvidia-pyindex; platform_machine == 'x86_64' nvidia-tensorrt == 8.4.1.5; platform_machine == 'x86_64' cuda-python == 11.7; platform_machine == 'x86_64' cython == 0.29.*; platform_machine == 'x86_64' -nvidia-cuda-runtime-cu11 == 2022.4.25; platform_machine == 'x86_64' -nvidia-cuda-runtime-cu117 == 11.7.*; platform_machine == 'x86_64' -nvidia-cublas-cu11 == 2022.4.8; platform_machine == 'x86_64' -nvidia-cublas-cu117 == 11.10.*; platform_machine == 'x86_64' -nvidia-cudnn-cu11 == 2022.5.19; platform_machine == 'x86_64' -nvidia-cudnn-cu116 == 8.4.*; platform_machine == 'x86_64' \ No newline at end of file +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' \ No newline at end of file