TensorRT安装

TensorRT安装

TensorRT-GA + Win0 / Ubuntu18.04 安装

---

---

前言

TensorRT有GA和EA两个版本，即稳定版和体验版，建议安装GA版本。

下载TensorRT时，根据nvcc -V查看CUDA版本选择相应的TensorRT，下面是TensorRT的下载链接：

https://developer.nvidia.com/tensorrt-getting-started

---

一、TensorRT+Win10安装

这里我下载的是 TensorRT-8.2.1.8.Windows10.x86_64.cuda-11.4.cudnn8.2.zip，解压后，将文件夹中的 lib 目录加入到系统环境变量 PATH 中

同时将 lib 文件夹下的*.dll和*.lib文件分别拷贝到 cuda 安装目录下的 bin 文件夹与lib/x64下，我这里是 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64

然后使用VS2019打开示例工程 D:\Software\TensorRT-8.2.1.8\samples\sampleMNIST\sample_mnist.sln 来验证下TensorRT安装是否正确。进入VS2019 后，打开sample_mnist 属性页

在 VC++目录-->可执行文件目录 中添加 tensorrt中的 lib 目录

然后在 C/C++ --> 常规 --> 附加包含目录中添加 tensorrt 中的 include 目录

最后，运行程序，可以得到