用pip安装gpu后找不到tensorflow

Question

我在 docker 图片上，所以我无法访问 docker 图片的 "outside"。我想安装支持 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)

之后我只得到输出

No module named tensorflow

如果我检查 pip 列表：

pip list | grep tensorflow

我得到输出：

tensorflow-gpu (1.0.1)

是简单的错误导入吗？

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

Device mapping: no known devices.

当然是因为不支持gpu。总而言之，如何让 tensorflow 的 GPU 版本与简单的 pip 安装和 1.0 以上的版本一起工作？

Answer 1

通过以下方式激活 tensorflow：

>>>source activate tensorflow

Answer 2

用

安装

conda install tensorflow-gpu

解决了所有问题。

用pip安装gpu后找不到tensorflow

Cannot find tensorflow after installing with pip for gpu

python

tensorflow

pip

cudnn