输入如下命令: 

python
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

若包含CPU信息,则使用了CPU。

若包含GPU信息,则使用了GPU。 

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐