tensorflow 2.x 和1.x 切換

如在安裝2以上的TensorFlow想使用1的版本,只需在文件開頭添加如下三行語句

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior() # 使用靜態圖模式運行以下代碼
assert tf.__version__.startswith('2.')
 

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