Tensorflow入门之 Hello World

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 tf


def 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 demo
    sess = tf.Session()
AttributeError: module 'tensorflow' has no attribute 'Session'

解决方案

• Tensorflow 版本导致
• 禁用TF2.x中的eager，否则会报错
• tf.Session() 改为 tf.compat.v1.Session()

完整代码

import tensorflow as tf


def demo():

    # 禁用TF2.x中的eager
    tf.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()

执行结果