08 2020 档案

神经网络训练时出现 无法获取卷积算法的问题 tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
摘要：使用训练环境： Tensorflow 2.2.0，CUDA 10.1，cuDNN 7.6.5。 当实例化VGG16卷积基时，出现了无法获取卷积算法的问题，提示可能因为cuDNN无法初始化。 打开代码文件在前面添加如下行，问题得以解决。 from tensorflow.compat.v1 import 阅读全文
posted @ 2020-08-09 16:47 BooTurbo 阅读(3436) 评论(1) 推荐(1) 编辑

点击右上角即可分享
微信分享提示