在Google Colab中使用TensorFlow遇阻,寻求解决方案
问题原因及解决办法
问题原因
你使用了TensorFlow 1.x的代码写法,但Google Colab默认搭载的是TensorFlow 2.x版本,两者API不兼容:
tf.ConfigProto和tf.Session是TF1.x的核心配置与会话类,TF2.x已移除这些API,改用即时执行模式(Eager Execution),无需显式创建Session。
解决办法
有两种可选方案:
方案1:适配TensorFlow 2.x写法
直接删除代码中关于ConfigProto和Session的部分,TF2.x默认开启即时执行,可直接编写模型逻辑。若需配置GPU,替换为以下代码:
import tensorflow as tf physical_devices = tf.config.list_physical_devices('GPU') tf.config.experimental.set_memory_growth(physical_devices[0], True)
方案2:切换到TensorFlow 1.x环境
在代码开头添加以下命令,强制使用TF1.x:
%tensorflow_version 1.x import tensorflow as tf
之后即可正常运行原TF1.x代码。
内容的提问来源于stack exchange,提问作者GRANDPA JOHN
相关产品推荐
相关产品推荐

