gpt4 book ai didi

Keras - 导入错误 : cannot import name 'CuDNNLSTM'

转载 作者:行者123 更新时间:2023-12-03 15:03:41 27 4
gpt4 key购买 nike

我正在尝试使用 CuDNNLSTM Keras 单元来提高循环神经网络的训练速度(文档 here)。

当我运行时:

from keras.layers import Bidirectional, CuDNNLSTM

我收到此错误:

ImportError: cannot import name 'CuDNNLSTM'



我的配置是 Keras 2.0.8、python 3.5、tensorflow-gpu 1.4.0(均由 Anaconda 管理)并且我安装了 CUDA 8.0 和 cudnn 6.0,这对于 tensorflow 的 nvidia 依赖项应该没问题( here )。我的代码设置使 Keras 有效地使用了 tensorflow 后端,并且除了以 CuDNN* 开头的层之外的每一层都可以正常工作。

任何人都知道此导入错误的来源?

最佳答案

对于 Tensorflow-2:您可以只使用没有激活功能的 LSTM,它会自动使用 CuDNN 版本

关于Keras - 导入错误 : cannot import name 'CuDNNLSTM' ,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47857437/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com