1. 环境准备

1.1 安装tensorrt

1.1.1 pip安装

pip install tensorrt 

1.1.2 压缩包安装

很可能会失败，最保险的方法是下载tensorRT的压缩包，比如：下载TensorRT-8.4.3.1
文件结构：


安装

pip install tensorrt-8.4.3.1-cp37-none-win_amd64.whl

配置环境变量

2. pt转engine

进入yolov5-7.0目录下（注意：必须是7.0以上，6.0的export.py不支持）

python export.py --weights E:\code\other\tph-yolov5-main\runs\train\exp\weights\last.pt --include engine --opset 12 --dynamic --imgsz 1536 --device 0