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 
    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()

执行结果

Tensorflow入门之 Hello World_第1张图片

你可能感兴趣的:(Python,Tensorflow,tensorflow,人工智能,python)