window安装tensorRT真的费劲,记录下遇到的问题,备忘!
1. 环境:
window10
TensorRT-7.0.0.11
cuda 10.2
cudnn 8.0.3
刚开始以为是TensorRT版本太低,7.0换成7.2.3,还是报一样的错误。
即环境
window10
TensorRT-7.2.3
cuda 10.2
cudnn 8.0.3
也不行。
2.环境
升级cuda版本,由cuda10.2升级到11.0。
报新的错误信息:
C:\source\rtSafe\cuda\cudaConvolutionRunner.cpp (483) - Cudnn Error in nvinfer1::rt::cuda::CudnnConvolutionRunner::executeConv
从错误信息看是cudnn的问题。
3.环境(正确环境)
可能是显卡太高端了,全面升级
tensorRT 7.2.3 (TensorRT-7.2.3.4.Windows10.x86_64.cuda-11.1.cudnn8.1.zip)
cuda11.1 (cuda_11.1.0_456.43_win10.exe)
cudnn8.1 (cudnn-11.2-windows-x64-v8.1.0.77.zip)
问题解决
注:mnist数据集 (网盘提取码:d0gc)