我试图使用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 *开头的层以外的其他所有层都可以正常工作。
有人知道此导入错误的来源吗?