用pip for gpu安装后找不到tensorflow

Question

我在一个docker镜像上，所以我不能访问docker镜像的“outside”。我想安装支持gpu的tensorflow：

^{1}$

cudnn和CUDA已安装并工作。旧版本（0.11）在映像中可用，并且可以与CUDA和cudnn一起运行，但我需要升级到版本1或更高版本。我有两个英伟达巨人。在

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

^{pr2}$

在这之后我只得到输出

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 

解决了所有问题。在