guilherme-pombo / resnet_tensorrt Goto Github PK

hi
when i run the eaxmple
there is a error
ModuleNotFoundError: No module named 'tensorrt.parsers'

coud you help me?
my tensorrt version is 7.0.0.1

[TensorRT] ERROR: runtime.cpp (16) - Cuda Error in allocate: 2
File "/usr/lib/python3.5/dist-packages/tensorrt/utils/_utils.py", line 283, in uff_to_trt_engine
assert(engine)
[TensorRT] ERROR: runtime.cpp (16) - Cuda Error in allocate: 2
[TensorRT] ERROR: Failed to create engine
Traceback (most recent call last):
File "/usr/lib/python3.5/dist-packages/tensorrt/utils/_utils.py", line 283, in uff_to_trt_engine
assert(engine)
AssertionError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "/home/wang/Downloads/Resnet_Tensorrt-master/tensorrt_wrapper.py", line 148, in
model_output_name=model_output_name)
File "/home/wang/Downloads/Resnet_Tensorrt-master/tensorrt_wrapper.py", line 40, in init
self.context = self.parse_uff_model(uff_path=uff_path)
File "/home/wang/Downloads/Resnet_Tensorrt-master/tensorrt_wrapper.py", line 60, in parse_uff_model
datatype=TRT_DATATYPE)
File "/usr/lib/python3.5/dist-packages/tensorrt/utils/_utils.py", line 291, in uff_to_trt_engine
raise AssertionError('UFF parsing failed on line {} in statement {}'.format(line, text))
AssertionError: UFF parsing failed on line 283 in statement assert(engine)

Process finished with exit code 1