python - 在 colab 中运行 jupyter 笔记本

Question

在 tensorflow 概率中运行 colab 示例时出现错误。

https://github.com/tensorflow/probability/blob/master/tensorflow_probability/examples/jupyter_notebooks/Bayesian_Gaussian_Mixture_Model.ipynb

尝试更改它，但我认为代码需要更新。

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 笔记本应该运行最新的更新。 我认为代码需要更新，我尝试更新配置但出现更多错误。

Answer 1

由于错误显示模块“tensorflow”没有属性“ConfigProto”，这意味着您需要导入它..如果它不起作用，则在虚拟环境中导入