确认自己的TensorFlow是CPU版本还是GPU版本

这是一个运行TensorFlow操作的类，他封装了操作和Tensor的计算环境 那怎么确认我们是安装的那个呢？ 记录设备指派情况 为了获取你的 operations 和 Tensor 被指派到哪个设备上运行, 用 log_device_placement 新建一个 session, 并设置为 True. 跑一个矩阵相乘 import tensorflow as tf # 新建一个 graph. a

>>阅读原文<<