使用pip install升级后,无法在GPU中配置Tensorflow?

时间:2016-11-12 00:25:33

标签: python tensorflow

我已将张量流从0.10升级到0.11,但是当我运行此代码时,我的张量流似乎只使用了CPU:

import tensorflow as tf
# 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)

我曾尝试在tensorflow的根目录中使用'./configure',但之后GPU仍未启用。我必须卸载整个tensorflow才能再次使用GPU吗?

0 个答案:

没有答案