检测自己tensorflow版本是使用CPU还是GPU

打开命令界面或者pycharm中输入:

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

检测自己tensorflow版本是使用CPU还是GPU

这里可以看到是cpu支持所以是cpu,如果显示gpu那就是gpu的