AssertionError:仅支持TensorFlow v1版本的报错解决咨询
解决TensorFlow版本不兼容问题
方案一:安装TensorFlow 1.x版本
当前代码明确要求仅支持TensorFlow 1.x,可卸载现有版本并安装指定1.x版本:
- 卸载当前TensorFlow:
pip uninstall tensorflow -y - 安装最新的TensorFlow 1.x稳定版(1.15.5):
pip install tensorflow==1.15.5 - 若需GPU支持,安装GPU版本:
注意:TensorFlow 1.15.5需搭配CUDA 10.0及对应cuDNN版本,需确保环境匹配。pip install tensorflow-gpu==1.15.5
方案二:修改代码兼容TensorFlow 2.x
利用TensorFlow 2.x的兼容模块,无需更换版本即可运行原代码:
将原代码替换为以下内容:
import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' # or any {'0', '1', '2'} import tensorflow.compat.v1 as tf tf.disable_v2_behavior() # 移除原断言语句,因已启用TF1兼容模式
启用兼容模式后,TensorFlow 2.x会模拟TF1的运行环境,原TF1风格的代码大多可正常执行。后续遇到TF1特有的API调用,均可通过tensorflow.compat.v1模块访问。
内容的提问来源于stack exchange,提问作者Pooooooorle
相关产品推荐
相关产品推荐

