环境配置:Tensorrt8.6.1 ,cuda11.6, cudnn8.6.0, torch1.13.1+cu116, torchvision0.14.1+cu116
1.先把cuda11.6, cudnn8.6.0, torch1.13.1+cu116, torchvision0.14.1+cu116这几个安装完之后
2.下载Tensorrt8.6.1,复制移动以下文件
将include中头文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include\
将lib中所有lib文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\lib\x64\
将lib中所有dll文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin\
3.去英伟达官网找到对应的zlibwapi.dll(依据电脑版本下载X64、X86版本的)文件,地址如下:Installation Guide - NVIDIA Docs,并将该dll拷贝到C:\Windows\System32文件夹下
4.在Tensorrt8.6.1/python里面有几个库,在cmd用pip install 库文件名。 来安装这些依赖库(python版本是3.10,所以安装对应的这三个库)
5.安装cuda-python库
pip install cuda-python
6.在cmd运行tensorrt测试案例:
7.运行过程中会显示少一些库,使用pip安装对应的库就OK了。运行成功如图所示