VS 2019配置tensorRT

配置TensorRT

1、下载TensroRT：链接：https://developer.nvidia.com/nvidia-tensorrt-8x-download

2、将下载的文件进行解压，并将TensorRT中的include文件移动到CUDA文件夹下的include中,并将TensortRT文件中的lib中的dll与lib文件分别放到cuda中的bin与lib\x64文件夹中

3、安装TensorRT，分别对这四个文件进行安装

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

4、下载cudnn64_8.dll，并将该文件放到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin 目录下

5、运行代码

import torch
import tensorrt as trt

print(trt.__version__)
print(torch.cuda.is_available())

6、运行示例，修改指定地方

7、下载cudnn：https://developer.nvidia.com/rdp/cudnn-download，将三个文件夹下的文件复制到NVIDIA GPU Computing Toolkit下对应的文件夹下

8、下载zlibwapi.dll文件，并将文件放在system32文件夹下，运行结果

若有误，请指正（谢谢）