相关疑难解决方法(0)

Keras-ImportError:无法导入名称“ CuDNNLSTM”

我试图使用CuDNNLSTM Keras细胞,以提高训练速度的回归神经网络(DOC 这里)。

当我跑步时:

from keras.layers import Bidirectional, CuDNNLSTM
Run Code Online (Sandbox Code Playgroud)

我收到此错误:

ImportError:无法导入名称“ CuDNNLSTM”

我的配置是Keras 2.0.8,python 3.5,tensorflow-gpu 1.4.0(均由Anaconda管理),并且我同时安装了CUDA 8.0和cudnn 6.0,应该可以与tensorflow的nvidia依赖关系(在此处)配合使用。我的代码设置使Keras有效地使用了tensorflow后端,除以CuDNN *开头的层以外的其他所有层都可以正常工作。

有人知道此导入错误的来源吗?

keras cudnn keras-layer

3
推荐指数
2
解决办法
4007
查看次数

标签 统计

cudnn ×1

keras ×1

keras-layer ×1