2018-10-31

检测tensorflow是否使用gpu进行计算

import tensorflow as tf

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

如果有些时候tensorflow无法使用GPU卡时，可以尝试cuda，cudnn和tensorflow重新安装。