tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize,

https://blog.csdn.net/zhangpeterx/article/details/89175991

二、解决方法

因为我一开始是直接在Pycharm里安装的tensorflow-gpu库,个人感觉应该是缺少了相关的库安装导致的。
故我使用conda再次安装一下tensorflow-gpu,

conda install tensorflow-gpu

然后问题就解决了。

猜你喜欢

转载自www.cnblogs.com/zb-ml/p/12811844.html
今日推荐