apps/frigate-rockchip
1
0
Fork 0
mirror of https://github.com/MarcA711/frigate-rockchip.git synced 2025-10-06 09:16:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update library loading for tensorrt (#5087)

Browse Source 
* Update library loading for tensorrt

* Add symlink to libnvrtc
This commit is contained in:
Nate Meyer
2023-01-14 14:14:27 -05:00
committed by GitHub
parent 19d17c8c81
commit 60b2315028
4 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/detectors/plugins/tensorrt.py
View File

@@ -75,12 +75,6 @@ class TensorRtDetector(DetectionApi):
def _load_engine(self, model_path):
try:
ctypes.cdll.LoadLibrary(
"/usr/local/lib/python3.9/dist-packages/nvidia/cuda_runtime/lib/libcudart.so.11.0"
)
ctypes.cdll.LoadLibrary(
"/usr/local/lib/python3.9/dist-packages/tensorrt/libnvinfer.so.8"
)
trt.init_libnvinfer_plugins(self.trt_logger, "")
ctypes.cdll.LoadLibrary("/trt-models/libyolo_layer.so")