TensorFlow Action（开山使用篇）

一、TensorFlow安装：

　　使用pip install tensorflow安装CPU版；

　　或使用pip install tensorflow-gpu==1.2.1指定版本安装GPU版。

二、TensorFlow测试样例：

import tensorflow as tf a = tf.constant([1.0, 2.0], name="a") b = tf.constant([3.0, 4.0], name="b") add = a + b sess = tf.Session() sess.run(add)

　　使用python的import操做加载TensorFlow；

　　使用tf.constant函数定义两个常量tensor，并将两个tensor相加；

　　使用tf.Session函数生成一个会话（session）并经过这个session来计算结果；

　　以上若运行无误，则说明TensorFlow安装成功。

三、TensorFlow指定GPU占用 示例：

import os import tensorflow as tf os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "2" config = tf.ConfigProto() # 设置定量的GPU使用量,占用GPU50%的显存 # config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 设置最小的GPU使用量，自动增加：按需求分配显存
config.gpu_options.allow_growth = True # 打印详细日志信息
config.log_device_placement = True a = tf.constant([1.0, 2.0], name="a") b = tf.constant([3.0, 4.0], name="b") add = a + b with tf.Session(config=config) as sess: sess.run(add)

　　使用os.environ指定具体的GPU，指定GPU占用可参考TensorFlow指定GPU使用及监控GPU占用状况

　　使用tf.ConfigProto()函数设置config设定GPU使用量及tf的其余属性参数，并在tf.Session()中指定该config。