Tensorflow入门之 Hello World
简介
Tensorflow 是 Google 开源的深度学习框架,来自于 Google Brain 研究项目,在 Google 第一代分布式机器学习框架 DistBelief 的基础上发展起来。
Tensorflow 的官方网址 http://www.tensorflow.org
Tensorflow 的 GitHub 网址 https://github.com/tensorflow/tensorflow
Hello World
启动 main 方法,若成功输出b’Hello, Tensorflow!’,则表明Tensorflow安装成功。
import tensorflow as tfdef demo():hello = tf.constant('Hello, Tensorflow!')sess = tf.Session()print(sess.run(hello))sess.close()if __name__ == '__main__':demo()
报错信息
Traceback (most recent call last):File "D:/Greenarrow/Workspace/FaceLogin/api/demo.py", line 13, in <module>demo()File "D:/Greenarrow/Workspace/FaceLogin/api/demo.py", line 7, in demosess = tf.Session()
AttributeError: module 'tensorflow' has no attribute 'Session'
解决方案
- Tensorflow 版本导致
- 禁用TF2.x中的eager,否则会报错
tf.Session()
改为tf.compat.v1.Session()
完整代码
import tensorflow as tfdef demo():# 禁用TF2.x中的eagertf.compat.v1.disable_eager_execution()hello = tf.constant('Hello, Tensorflow!')sess = tf.compat.v1.Session()print(sess.run(hello))sess.close()if __name__ == '__main__':demo()