sess.run()这行命令前面，加上如下：
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))
终端的device mapping
Device mapping:
/job:localhost/replica:0/task:0/device:GPU:0 -> device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1
MatMul: (MatMul): /job:localhost/replica:0/task:0/device:GPU:0
a: (Const): /job:localhost/replica:0/task:0/device:GPU:0
b: (Const): /job:localhost/replica:0/task:0/device:GPU:0
最后
以上就是开朗云朵最近收集整理的关于怎样知道tensorflow计算用的是CPU还是GPU的全部内容，更多相关怎样知道tensorflow计算用内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复