yolov5的TensorRT部署【tensorrt+cudnn@主机】

1、版本问题

（1）TensorRT的ONNX解释器是针对Pytorch版本编译的，如果版本不对应可能导致转模型时出现错误。

（2）主机环境：pytorch1.8.1+cuda11.1

2、Pytorch2ONNX

主机上exp305权重可用，尝试转ONNX，命令如下：

python export.py --weights runs/train/exp305/weights/best.pt --img 640 --include engine onnx 

成功导出best.onnx到runs/train/exp305/best.onnx

3、YOLO2TensorRT

export可以直接导出tensorrt，参数是--include engine，自动下载了tensorrt-0.0.1

包不能用：

No module named 'tensorrt'

安装下载TensorRT(这里是错误方法，可直接跳到下面正确方法)

参照别人的版本：

 尝试更换tensorrt版本：

pip uninstall tensorrt

因为看到英伟达官网上只有8.2有ubuntu20.02版本，所以下载了这个版本

下载安装了trt8.2，仍然不行，no module named 'tensorrt'：

用dpkg安装：

还是不行。

版本不对需要不同的依赖：