检测自己tensorflow版本是使用CPU还是GPU
打开命令界面或者pycharm中输入:
python
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
这里可以看到是cpu支持所以是cpu,如果显示gpu那就是gpu的
打开命令界面或者pycharm中输入:
python
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
这里可以看到是cpu支持所以是cpu,如果显示gpu那就是gpu的