TensorRT7.0学习:由于找不到cudnn64_7.dll (win+vs2017+CUDA10.2)

正在安装TensorRT,配好了库,但出了这么一个报错,如题目中。

我想运行附带的手写数字分类例程
TensorRT7.0学习:由于找不到cudnn64_7.dll (win+vs2017+CUDA10.2)_第1张图片
这个错误,我找了很久,也没发现相关的资料。 起初以为是路径include的少了,但后来发现,并非如此。因为在C盘的某个文件夹下,我有cudnn64_8.dll文件。这正是问题所在。也就是,cudnn版本不对。

原来的cudnn版本:cudnn-10.2-windows10-x64-v8.2.1.32
起始,这个v8.2版本对于pytorch的模型训练而言,没什么版本不对应。但TensorRT有特定的版本要求。我的TensorRT是7.0的,需要的cudnnv7
我后来下载的新版本:cudnn-10.2-windows10-x64-v7.6.5.32
将cudnn的文件更换文件夹后,问题OK

你可能感兴趣的:(学习,深度学习,pytorch)