Error : Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so

every blog every motto: Brevity is the soul of wit

0. 前言

在用tensorflow 进行训练时，添加了如下代码：

Error : Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

主要原因是版本问题，还有可能是没有安装cuda / cudnn ,具体可在pycharm的设置中查看具体版本。
这里推荐如下解决办法。

1. 解决办法

1. 所需版本

NVIDIA驱动 需要大于410.X
cuda 版本需要是10.0 （图鉴10.0.130）
cudnn版本需要大于7.4.1 （推荐7.6.0）

2. 查看conda中可用的版本

2.1 查看cuda

conda search cudatoolkit

2.2 查看cudnn

conda search cudnn

3. 正确安装Tensorflow cpu/gpu

3.1 安装tensorflow cpu

pip install tensorflow==2.0.0

3.2 安装NIVIDA驱动

官网下载

3.3 安装cudnn cuda

conda install cudatoolkit=10.0.130
conda instal cudnn=7.6.0

3.4 安装tensorflow gpu

pip isntall tensorflow-gpu

参考来源

[1] https://github.com/tensorflow/tensorflow/issues/24828
[2] https://study.163.com/course/courseLearn.htm?courseId=1004573006#/learn/video?lessonId=1279234034&courseId=1004573006