模块'tensorflow'没有属性'GPUOptions'

Question

我正在使用具有GTX 960M的Dell笔记本电脑，并且安装了tensorflow 2.0 alpha。我曾经使用1.5版本，并且可以与tf.GPUOptions一起使用，但是在此版本中，这会导致error: AttributeError: module 'tensorflow' has no attribute 'GPUOptions' 如果我用这个版本做错了事，有人可以帮我

  

回溯（最近通话最近）：         文件“ prepare_training.py”，第20行，在           gpu_option = tf.GPUOptions（per_process_gpu_memory_fraction = 0.333）       AttributeError：模块“ tensorflow”没有属性“ GPUOptions”

Answer 1

Tensorflow 2.x与1.x相比发生了重大变化。

根据官方通讯，

  

tf.contrib将被从TensorFlow核心存储库和构建过程中删除。 TensorFlow的contrib模块已经超出了单个存储库中可以维护和支持的范围。较大的项目最好单独维护，而较小的扩展将逐步扩展到TensorFlow核心代码。已经成立了一个特殊利益集团（SIG），以维护和进一步发展一些今后更重要的贡献项目。如果您有贡献，请参与this RFC。

如果要使用tensorflow 1.x函数/方法，则tensorflow 2.x中保留有一个兼容性模块。

tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction)