以张量流概率运行 colab 示例我收到错误。
尝试更改它，但我认为代码需要更新。
def session_options(enable_gpu_ram_resizing=True):
"""Convenience function which sets common `tf.Session` options."""
config = tf.ConfigProto()
config.log_device_placement = True
if enable_gpu_ram_resizing:
# `allow_growth=True` makes it possible to connect multiple colabs to your
# GPU. Otherwise the colab malloc's all GPU ram.
config.gpu_options.allow_growth = True
return config
def reset_sess(config=None):
"""Convenience function to create the TF graph and session, or reset them."""
if config is None:
config = session_options()
tf.reset_default_graph()
global sess
try:
sess.close()
except:
pass
sess = tf.InteractiveSession(config=config)
reset_sess()
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-8-8f3c54a802d4> in <module>()
21 sess = tf.InteractiveSession(config=config)
22
---> 23 reset_sess()
1 frames
<ipython-input-8-8f3c54a802d4> in reset_sess(config)
12 """Convenience function to create the TF graph and session, or reset them."""
13 if config is None:
---> 14 config = session_options()
15 tf.reset_default_graph()
16 global sess
<ipython-input-8-8f3c54a802d4> in session_options(enable_gpu_ram_resizing)
1 def session_options(enable_gpu_ram_resizing=True):
2 """Convenience function which sets common `tf.Session` options."""
----> 3 config = tf.ConfigProto()
4 config.log_device_placement = True
5 if enable_gpu_ram_resizing:
AttributeError: module 'tensorflow' has no attribute 'ConfigProto'
没有错误。colab 笔记本应使用最新更新运行。我认为代码需要更新，我试图用更多错误更新配置。