azaion/annotations
1
0
Fork 0
mirror of https://github.com/azaion/annotations.git synced 2026-04-22 10:56:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

import Tensorrt not in compile time in order to dynamically load tensorrt only if nvidia gpu is present

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-04-30 23:08:53 +03:00
parent ae83bc8542
commit 1c4bdabfb5
4 changed files with 12 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Inference/inference.pyx
+1 -1
View File
@@ -16,7 +16,7 @@ from hardware_service cimport HardwareService
from security cimport Security
if HardwareService.has_nvidia_gpu():
from tensorrt_engine cimport TensorRTEngine
from tensorrt_engine import TensorRTEngine
else:
from onnx_engine import OnnxEngine