确定自己的TensorFlow是CPU还是GPU的版本
更新了一次TensorFlow之后,才发现更成了CPU版,然后GPU版又存在,无比的难受。因为无法确定现在运行的TensorFlow到底是CPU版还是GPU版。然后找到了解决方案:
在Python环境中输入:
import numpy
import tensorflow as tf
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[...
转载
2018-11-14 21:39:15 ·
2113 阅读 ·
0 评论