1

那是我第二次经历 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))

（来源）

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

任何帮助都会很有用，在此先感谢您。

亲切的问候，康尼

4

1 回答 1

1

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

于 2017-06-28T14:01:44.533 回答