Tensorflow GPU安装正确,但Anaconda在CPU上运行它

时间:2017-06-28 13:35:50

标签: python tensorflow windows-10 tensorflow-gpu

这是我第二次完成tensorflow(gpu)的整个安装过程。这次我在cmd activate tensorflow中运行然后如果我在python解释器中输入以下测试代码:

# Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print(sess.run(c))

The Source

它成功识别GPU并在其上运行矩阵函数。当我打开Anaconda Spyder IDE时,以相同的方式运行相同的测试,解释器说代码正在CPU上运行。我该如何解决这个问题?可能导致什么呢? 我在Win 10上运行。

任何帮助都会很有用,请提前感谢。

亲切的问候, Konny

1 个答案:

答案 0 :(得分:1)

这不是地球上最好的解决方案,但以下对我有用:  1.打开cmd。  2.输入activate tensorflow  3.输入spyder  4.让它加载一切。 现在上面的测试表明,GPU上确实存在张量流。 这不是一个好的解决方案,因为每个spyder的开放必须以这种方式发生。现在这个工作。是的,它默认指向错误的tensorflow环境。我的问题是,我不知道cmd中运行的那个在我的计算机上的位置。当我设法解决这个问题时,我会更新答案。

相关问题