Android Studio中Kotlin变量声明报错求助,疑版本不匹配
报错原因及解决方案
红色下划线处的tf.keras.layers.CuDNNLSTM API在TensorFlow 2.x及后续版本中已被移除,这是导致报错的直接原因。YouTube视频中的代码基于TensorFlow 1.x版本编写,和你当前使用的TF版本不兼容。
解决办法:
推荐方案:替换为标准LSTM层
TensorFlow 2.x的tf.keras.layers.LSTM会自动检测GPU环境,当系统配置了兼容的CUDA和CuDNN时,会自动启用CuDNN加速,无需显式调用CuDNNLSTM。直接将代码中的CuDNNLSTM替换为LSTM即可,示例:# 原代码 tf.keras.layers.CuDNNLSTM(64, return_sequences=True) # 修改后 tf.keras.layers.LSTM(64, return_sequences=True)若替换后出现激活函数相关警告,可显式指定
recurrent_activation='sigmoid',这是原CuDNNLSTM默认的激活方式。备选方案:降级到TensorFlow 1.x
如果你必须和视频代码完全一致,可通过pip安装旧版本TF:pip install tensorflow==1.15但不推荐此方案,因为TensorFlow 1.x已停止官方维护,后续会面临安全和功能更新问题。
内容的提问来源于stack exchange,提问作者NInad Kulkarni
相关产品推荐
相关产品推荐

