TensorFlow（一）

一、Hello World

1.只安装CPU版，TensorFlow1.14.0版本代码

# import tensorflow as tf
import tensorflow.compat.v1 as tf
import os

# os.environ["TF_CPP_MIN_LOG_LEVEL"] = '1' # 默认，显示所有信息
os.environ["TF_CPP_MIN_LOG_LEVEL"] = '2'  # 只显示 warning 和 Error
# os.environ["TF_CPP_MIN_LOG_LEVEL"] = '3' # 只显示 Error


if __name__ == '__main__':
    hello = tf.constant('Hello, TensorFlow!')  # 定义一个常量
    sess = tf.Session()  # 建立一个session
    print(sess.run(hello))  # 通过 session里面的 run来运行结果
    sess.close()  # 关闭 session

二、其他问题

1.设置运行窗口显示信息

import os

# os.environ["TF_CPP_MIN_LOG_LEVEL"] = '1' # 默认，显示所有信息
os.environ["TF_CPP_MIN_LOG_LEVEL"] = '2'  # 只显示 warning 和 Error
# os.environ["TF_CPP_MIN_LOG_LEVEL"] = '3' # 只显示 Error

2.版本，GPU是否使用

version = tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version:",version,"\nuse GPU",gpu_ok)

3.如果同时安装了两个版本，默认下运行gpu版，如想运行cpu版，可代码中设置如下：

with tf.Session() as ses:
    with tf.device("/cpu:0"): # "/gpu:0"是GPU
        matrix1=tf.constant([[3.,3.]])
        matrix2=tf.constant([[2.],[2.]])
        product=tf.matmul(matrix1,matrix2)