我正在尝试使用 CuDNNLSTM Keras 单元来提高循环神经网络的训练速度（文档here https://keras.io/layers/recurrent/#cudnnlstm).
当我跑步时：
from keras.layers import Bidirectional, CuDNNLSTM
我收到此错误：
导入错误：无法导入名称“CuDNNLSTM”
我的配置是 Keras 2.0.8、python 3.5、tensorflow-gpu 1.4.0（全部由 Anaconda 管理），并且我安装了 CUDA 8.0 和 cudnn 6.0，这应该可以满足 tensorflow 的 nvidia 依赖项（here https://github.com/tensorflow/tensorflow/releases）。我的代码设置使 Keras 有效地使用张量流后端，并且除了以 CuDNN* 开头的层之外的每一层都工作正常。
有人知道这个导入错误的根源吗？
对于 Tensorflow-2：您可以只使用没有激活功能的 LSTM，它会自动使用 CuDNN 版本
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
