安装好tensorflow2.0之后,当使用Session时,报错AttributeError: module 'tensorflow' has no attribute 'Session':
源代码:
1 2 3 4 5 6 7 8 9 | import tensorflow as tf import os os.environ[ "CUDA_VISIBLE_DEVICES" ] = "0" a = tf.constant( 2 ) b = tf.constant( 3 ) with tf.Session() as sess: print ( "a:%i" % sess.run(a), "b:%i" % sess.run(b)) print ( "Addition with constants: %i" % sess.run(a + b)) print ( "Multiplication with constant:%i" % sess.run(a * b)) |
错误信息:
错误的意思是tensortflow模块没有Session属性,后来查阅资料发现,tensorflow2.0版本中的确没有Session这个属性,如果安装的是tensorflow2.0版本又想利用Session属性,可以将tf.Session()更改为:
这个方法可以解决此类问题,不仅仅适用于Session属性。
再次运行时,程序又报了另一个错误:
查阅资料发现,原因是2.0与1.0版本不兼容,在程序开始部分添加以下代码:
1 | tf.compat.v1.disable_eager_execution() |
就可以正常运行了。
tensorflow的官网对disable_eager_execution()方法是这样解释的:
1 | This function can only be called before any Graphs, Ops, or Tensors have been created. <br>It can be used at the beginning of the program for complex migration projects from TensorFlow 1.x to 2.x . |
翻译过来为:此函数只能在创建任何图、运算或张量之前调用。它可以用于从TensorFlow 1.x到2.x的复杂迁移项目的程序开头。
更新:
找到了一个更简单的方法,在引用tensorflow时,直接用:
1 | import tensorflow.compat.v1 as tf |
一劳永逸的方法。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)