Tensorflow升级后Failed to get convolution algorithm. This is probably because cuDNN

分类: 文章 • 2024-02-28 20:56:58

系统原始环境：

1. tensorflow：2.0.alpha

2. OS：Win10

3. Cuda：10.0.130

4. cudnn：7.5.0

将tensorflow升级至2.0正式版后运行程序报错：Failed to get convolution algorithm. This is probably because cuDNN···

解决方案：从官网下载最新版本的cudnn，此处使用的是cudnn7.6.5，更新cudnn后程序成功运行。

Tensorflow升级后Failed to get convolution algorithm. This is probably because cuDNN

PS：若不知道自己安装的cudnn版本是什么版本，查看

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\cudnn\include

路径下，cudnn.h中的宏定义：

Tensorflow升级后Failed to get convolution algorithm. This is probably because cuDNN

如上图所示，我的版本是7.6.5