如何检测tensorflow是否使用CPU还是GPU计算
输入如下命令:pythonimport tensorflow as tfsess = 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。
更多推荐



所有评论(0)