首先确认自己的CUDA版本,作者版本是11.6
1.现在下载匹配的 cuDNN,cuDNN Archive | NVIDIA DevelopercuDNN 8.4(与 TensorRT 8.4/8.5 兼容性较好)
2.将下载好的cuDNN解压,将里面的lib,include,bin文件复制粘贴到cuda中相同的位置
3.下载TensorRT Log in | NVIDIA Developer 版本 TensorRT 8.4 GA Update 2 for x86_64 Architecture 同样的操作将lib,include,bin文件复制粘贴到cuda中
之后测试一下
import tensorrt as trt
print(trt.__version__)