最近公司配置了新的台式机,显卡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()
如果还有其他问题,可放评论区一起讨论