输入如下命令:
python
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
若包含CPU信息,则使用了CPU。
若包含GPU信息,则使用了GPU。
输入如下命令:
python
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
若包含CPU信息,则使用了CPU。
若包含GPU信息,则使用了GPU。