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,
相关文章
相关标签/搜索