2022-05-24 1-1 WIN10下安装TensorRT 8.4_EA

1.环境描述

win10下 cuda10.2、cudnn-10.2_8.3.2.44、torch1.8.1、torchvision0.9.1、python3.7

百度盘：链接：https://pan.baidu.com/s/14ByxfmkkVJ4RzpKLC0cA4Q  提取码：iivv

cuda和cudnn一定要版本号为以上版本，不然onnx转engine时会报错，我已踩坑，如下图报错表及官方给的版本对照表。

2.安装步骤

1)将下载解压好的TensorRT-8.4.0.6文件夹中include移动到cuda安装目录下（C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2）；

2)TensorRT-8.4.0.6中lib文件夹dll文件移动到cuda安装目录bin下，lib文件移动到cuda安装目录lib\x64下;

3)pip安装whl文件。一共需要安装4个whl文件，分别在TensorRT-8.4.0.6中graphsurgeon、onnx_graphsurgeon、python、uff文件夹内。（注意：python文件夹中只安装对应python版本的whl文件。）

3.测试

我是在anaconda prompt中直接import tensorrt来测试。

import tensorrt as trt

print(trt.__version__)

后续更新yolov5+tensorrt。