AttributeError: модуль tenorrt не имеет атрибута infer

Question

Я хочу использовать tenorRT для развертывания моей модели pytorch, но у меня проблема при импорте tenorRT.

Traceback (most recent call last):
  File "/home/fffan/fffan_files/Experiment/pytorch-tensorrt-master/model2trt.py", line 10, in <module>
      import calib as calibrator
  File "/home/fffan/fffan_files/Experiment/pytorch-tensorrt-master/calib.py", line 13, in <module>
      class PythonEntropyCalibrator(trt.infer.EntropyCalibrator):
  AttributeError: module 'tensorrt' has no attribute 'infer'

Код выглядит следующим образом: tenorRT - 5.1.5. Кто знает, подскажите пожалуйста. Большое спасибо!