一、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)
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:TensorFlow1.0版 - Python技术站