Tutorial--怎麼看本身安裝的Tensorflow是GPU版本的仍是CPU版本的

例子1: 爲了獲取你的 operations 和 Tensor 被指派到哪一個設備上運行, 用log_device_placement 新建一個 session, 並設置爲 True來記錄設備指派狀況。python 例子(矩陣相乘):web import tensorflow as tf # 新建一個 graph a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0,
相關文章
相關標籤/搜索