原文:神经网络训练时出现 无法获取卷积算法的问题 tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

使用训练环境: Tensorflow . . ,CUDA . ,cuDNN . . 。 当实例化VGG 卷积基时,出现了无法获取卷积算法的问题,提示可能因为cuDNN无法初始化。 打开代码文件在前面添加如下行,问题得以解决。 ...

2020-08-09 16:47 1 2460 推荐指数:

查看详情

Failed to get convolution algorithm解决

蒸腾了两天,终于搞定了 是cudnn版本的问题 更新cudnn的时候,首先要删除/usr/local/cuda-10.0/targets/x86_64-linux/lib路径下所有之前cudnn版本的库,sudo rm libcudnn* 然后 将下载的新版本的cudnn拷贝到 ...

Wed Feb 26 00:32:00 CST 2020 0 1066
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM