在卷积神经网络中无法调用cudnn
在利用tf2.0搭建卷积神经网络时,一直无法运行,显示如下错误:
Failed to get convolution algorithm. This is probably because cuDNN failed 。
[[node sequential/conv2d/Conv2D (defined at .....
网上搜索了很多,大多数都是都是说有可能是显存不够的问题,我也朝着这个方向,进行了一些修正,但发现无法解决问题。
于是我又把目光转到报错上,经过很多坑之后,终于找到了问题的源头:
可以看到我的cudnn是直接放在CUDA的目录下,这是我安装tf2.0的时候放入的。
修改方法
将cudnn中的所有文件,放入CUDA中对应的文件夹内,即可修正。
注意事项
我在此之前,排除的坑:
1.版本对应
2.显存足够
3.全连接神经网络可运行
如果没在此类,此方法不一定有效。