当前位置: 代码迷 >> 综合 >> TensorFlow1.0版本keras代码转TensorFlow2.4不兼容问题的解决方法
  详细解决方案

TensorFlow1.0版本keras代码转TensorFlow2.4不兼容问题的解决方法

热度:11   发布时间:2023-12-03 22:08:58.0

最近公司配置了新的台式机,显卡RTX3090,喜忧参半,喜是可以用上高级货了,忧是我之前写的代码都是在TensorFlow_gpu1.14上。RTX3090上搭建的TensorFlow2.4.0和Keras2.4.3版本,很多地方改动很多,出现各种不兼容问题,导致一次次运行不成功。

其一:module 'tensorflow' has no attribute 'get_default_graph'

放弃使用自己安装的Keras模块,而是使用TensorFlow自带的Keras库

比如下面,在Keras前面加上TensorFlow

尽量全部改过来,包括yolo3文件夹内的model.py文件和自己写的代码

 

其二:如果还有其他问题,试试这个:

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

如果提示compat不存在,试试这个:

import tensorflow._api.v2.compat.v1 as tf
tf.disable_v2_behavior()

如果还有其他问题,可放评论区一起讨论

  相关解决方案