TensorRT5.1.5.0 实践 Pytorch2Onnx ,Onnx2TensorRT(python)
TensorRT5.1.5.x的关于onnx的demo只有两个,一个是c++的introductory_parser_samples,一个是python的ObjectDetectionWithTheONNXTensorRTBackendInPython。参考了各方资料,这篇文章是记录+实践最后成功的环境:Ubuntu16.06,python3.5.2,torch1.0.0,torchvision0