tensorflow——Could not set cudnn filter descriptor:CUDNN

xiaoxiao2021-02-28  39

这个问题困扰我好几天,后来受到这篇博客的启发,我发现是我代码的问题。

我计算batch的个数用的是原始数据train_data,但我实际用的时候只用了一半的数据train_data1,导致我的batch_times在train_data1后面的索引找不到数据,batch就是空的

就出现这样的错误,Could not set cudnn filter descriptor:CUDNN_STATUS_BAD_PARAM

以后写代码的时候一定要细心,养成好习惯

下面附上cudnn更新版本的代码

conda install cudnn==5.1(等号后面写要升级的版本号)

安装成功后会自动更新软链接的

附上我参考的博客地址:https://blog.csdn.net/jiruiyang/article/details/77968226

转载请注明原文地址: https://www.6miu.com/read-2627249.html

最新回复(0)