概述
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计算用的是CPU还是GPU所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复