這是一個執行TensorFlow操作的類，他封裝了操作和Tensor的計算環境

那怎麼確認我們是安裝的那個呢？

記錄裝置指派情況 為了獲取你的 operations 和 Tensor 被指派到哪個裝置上執行, 用 log_device_placement 新建一個 session, 並設定為 True.

跑一個矩陣相乘

import tensorflow as tf
# 新建一個 graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# 新建session with log_device_placement並設定為True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# 執行這個 op.
print (sess.run(c))
#任務完成 關閉會話
sess.close()
 

然後你會看到

我這裡是CPU版本