在运行程序时遇到一个问题
控制台显示错误:
tensorflow.python.framework.errors_impl.UnknownError: 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.
Function call stack:
keras_scratch_graph
原因 :
一 :安装的tensorflow版本和keras版本不匹配。
查看tensorflow和keras对应的版本网址:
https://docs.floydhub.com/guides/environments/
利用以下命令行安装相对应的版本即可:
pip install -i http: