[英]Running jupyter notebooks in colab
在张量流概率中运行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笔记本应使用最新更新运行。 我认为代码需要更新,我尝试使用更多错误来更新配置。
由于错误显示模块'tensorflow'没有属性'ConfigProto',这意味着您需要导入它..如果它不起作用,则导入虚拟环境
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.