一、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)