一、安装包文件准备
设备及软件版本
准备开发环境时,需要确定安装TensorRT版本,再反过来确定cudnn版本,CUDA版本,这样就不会有问题了,免得下载版本不对又得重新下载。
下载地址:
TensorRT:https://developer.nvidia.cn/nvidia-tensorrt-download
cudnn:https://developer.nvidia.cn/rdp/cudnn-archive
CUDA:https://developer.nvidia.cn/cuda-toolkit-archive
或者下载
百度网盘:https://pan.baidu.com/s/17K-snlqT2yD9ZIQ-GXEJOg 提取码:7ad1
- cuda_11.1.1_456.81_win10.exe
- cudnn-11.2-windows-x64-v8.1.1.33.zip
- TensorRT-8.0.3.4.Windows10.x86_64.cuda-11.3.cudnn8.2.zip
二、安装
- 首先安装CUDA,默认安装路径,直接下一步即可
- 解压缩
cudnn-11.2-windows-x64-v8.1.1.33.zip
选中全部(include、lib、bin)复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
目录下即可(自动合并include、lib、bin对应文件夹文件) - 解压缩
TensorRT-8.0.3.4.Windows10.x86_64.cuda-11.3.cudnn8.2.zip
分别将include、lib、bin三个目录中的内容复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
对应的include、lib/x64、bin目录下即可
三、测试
- 用VS2019 打开
TensorRT-8.0.3.4\samples\sampleMNIST\sample_mnist.sln
- 右键工程 ->重新生成
- 进入TensorRT-8.0.3.4\bin,用cmd执行sample_mnist.exe --datadir=path/to/data/mnist/
- 如果显示下图执行成功则说明tensorRT 配置成功
- 错误解决:找不到如下动态库,
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64
下如下文件复制到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
目录下。