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

This reverts commit 1c4bdabfb5.
This commit is contained in:
Alex Bezdieniezhnykh
2025-04-30 23:32:03 +03:00
parent 1c4bdabfb5
commit cf01e5d952
4 changed files with 20 additions and 12 deletions
+11 -3
View File
@@ -16,9 +16,17 @@ cdef class TensorRTEngine(InferenceEngine):
cdef object stream
@staticmethod
cdef get_gpu_memory_bytes(int device_id)
cpdef tuple get_input_shape(self)
@staticmethod
cdef get_engine_filename(int device_id)
cpdef int get_batch_size(self)
@staticmethod
cdef convert_from_onnx(bytes onnx_model)
cpdef run(self, input_data)
cdef tuple get_input_shape(self)
cdef int get_batch_size(self)
cdef run(self, input_data)