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

打开命令界面或者pycharm中输入: python import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 这里可以看到是cpu支持所以是cpu,如果显示gpu那就是gpu的
相关文章
相关标签/搜索