確認自己的TensorFlow是CPU版本還是GPU版本

這是一個運行TensorFlow操作的類，他封裝了操作和Tensor的計算環境 那怎麼確認我們是安裝的那個呢？ 記錄設備指派情況 爲了獲取你的 operations 和 Tensor 被指派到哪個設備上運行, 用 log_device_placement 新建一個 session, 並設置爲 True. 跑一個矩陣相乘 import tensorflow as tf # 新建一個 graph. a

>>阅读原文<<