使用pip for gpu安装后找不到tensorflow

Question

我在docker映像上，因此无法访问docker映像的“外部”。 我想安装使用gpu支持的tensorflow：

pip install tensorflow-gpu 

cudnn和CUDA已安装并正在运行。 映像中提供了旧版本（0.11），并且可以很好地与CUDA和cudnn一起运行，但是我需要升级到版本1或更高版本。 我有两个Nvidia Titans。

使用显示的pip命令后，我使用以下脚本查看是否启用了GPU支持，并查看nvidia-smi：

import tensorflow as tf


# Creates a graph.
with tf.device('/gpu:0'):
  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的模块

如果我使用以下方法检查点子列表：

pip list | grep tensorflow

我得到的输出：

tensorflow-gpu（1.0.1）

这是一个简单的错误导入吗？

如果我使用非gpu支持的安装pip install tensorflow则上面的代码将给出：

设备映射：无已知设备。

当然，这是由于没有GPU支持。 因此，总而言之，如何通过简单的pip安装和高于1.0的版本来获得Tensorflow的GPU版本？

Answer 1

用安装

conda install tensorflow-gpu 

解决了所有问题。

Answer 2

使用以下命令激活张量流：

>>>source activate tensorflow