文章目录[x]
- 1:环境版本信息：
- 2:解决办法
- 3:参考
在电脑上搭建的深度学习框架中引入卷积层时报错：
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
环境版本信息：
- python 3.7.9
- CUDA 10.0
- cuDNN 7.4.1
- Tensorflow 2.0.0
解决办法
-
CUDA,cuDNN,Tensorflow 版本不兼容。主要解决方法是 Tensorflow 版本太高，需要降版本。
-
GPU 内存不足。需要在代码最前面加几行代码实现对 GPU 进行按需分配。
最终解决办法： 将原有 cuDNN 7.4.1 版本更换为 7.6.5
删掉 CUDA bin,include,lib 文件夹中原有 cuDNN 7.4.2 版本拷贝的文件，再将 cuDNN 7.6.5 版本文件拷贝到相应文件夹中。
参考
解决 TF2.0: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize)