tensorflow 2.x 和1.x 切换

如在安装2以上的TensorFlow想使用1的版本,只需在文件开头添加如下三行语句

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior() # 使用静态图模式运行以下代码
assert tf.__version__.startswith('2.')
 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章