如何检测tensorflow是否使用CPU还是GPU计算

输入如下命令: 

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

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

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

猜你喜欢

转载自blog.csdn.net/weixin_42702666/article/details/88067012