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

系统原始环境:

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_第1张图片

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

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

路径下,cudnn.h中的宏定义:

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

 

你可能感兴趣的:(TensorFlow,tensorflow,cuda,cudnn)