Tensorflow2教程-基础CNN网络

[Baominchao]

您好，我在运行标题的程序时在命令行history = model.fit(x_train, y_train, batch_size=64, epochs=5, validation_split=0.1)遇到了这个问题： 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. TF版本是2.0正式版，电脑重启试过还是同样的问题。

[AndrewJSong]

cudNN重新装一下。检查CUDA版本10.0，cudnn找对应版本

发自我的 iPhone

在 2019年10月6日，19:46，Baominchao [email protected] 写道：

您好，我在运行标题的程序时在命令行history = model.fit(x_train, y_train, batch_size=64, epochs=5, validation_split=0.1)遇到了这个问题： 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. TF版本是2.0正式版，电脑重启试过还是同样的问题。

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or mute the thread.

[Baominchao]

谢谢您，启动Jupyter notebook的时候出现了这个：Loaded runtime CuDNN library: 7.5.0 but source was compiled with: 7.6.0. CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library. If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration. 然后我下载了7.6的版本，替换cudnn的版本就行了。再次感谢您。