这篇文章的前提是你已经安装好了cuda和cudnn。并且没有问题了。不会弄得看我的文章。
然后开始配置trt内容。
1 相关参数
cuda cuda_11.1.0_456.43_win10.exe
cudnn cudnn-11.1-windows-x64-v8.0.4.30
trt TensorRT-7.2.1.6.Windows10.x86_64.cuda-11.1.cudnn8.0.zip
MSVS 2019 社区版
2 添加环境变量
C:\TensorRT-7.2.1.6.Windows10.x86_64.cuda-11.1.cudnn8.0\TensorRT-7.2.1.6\lib
C:\TensorRT-7.2.1.6.Windows10.x86_64.cuda-11.1.cudnn8.0\TensorRT-7.2.1.6\bin
3 安装以来包
进到我指出的文件夹中,然后pip安装就行。
4 复制dll文件到cuda安装目录
复制lib下文件到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin 下面。