安装ONNX_tensorrt的python部分

首先静进入到onnx_tensorrt目录下

运行：python setup.py install

可能出现问题：

NvOnnxParser.h:26:21: fatal error: NvInfer.h: No such file or directory compilation terminated.

此时，你需要修改setup.py文件，首先，在电脑中搜索NvInfer.h，找到它的路径

  修改setup.py文件中的 代码，将注释掉的代码改为其下面的样子，便可以成功安装

# TRT_ROOT = os.getenv('TRT_ROOT')
# if TRT_ROOT == None:
#     INC_DIRS = []
# else:
#     print("TRT_ROOT: ",TRT_ROOT)
#     INC_DIRS = [TRT_ROOT + '/include']
TRT_ROOT = '/home/nvidia/TensorRT'
if TRT_ROOT == None:
    INC_DIRS = []
else:
    print("TRT_ROOT: ",TRT_ROOT)
    INC_DIRS = [TRT_ROOT + '/include']