mirror of
https://github.com/azaion/annotations.git
synced 2026-04-22 10:56:31 +00:00
Reapply "import Tensorrt not in compile time in order to dynamically load tensorrt only if nvidia gpu is present"
This reverts commit cf01e5d952.
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user