这是我第二次完成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上运行。
任何帮助都会很有用,请提前感谢。
亲切的问候, Konny
答案 0 :(得分:1)
这不是地球上最好的解决方案,但以下对我有用:
1.打开cmd。
2.输入activate tensorflow
3.输入spyder
4.让它加载一切。
现在上面的测试表明,GPU上确实存在张量流。
这不是一个好的解决方案,因为每个spyder的开放必须以这种方式发生。现在这个工作。是的,它默认指向错误的tensorflow环境。我的问题是,我不知道cmd中运行的那个在我的计算机上的位置。当我设法解决这个问题时,我会更新答案。