Question

我在码头图像上，所以我无法访问泊坞窗图像的“外部”。我想安装使用gpu支持的tensorflow：

4/10/2017

cudnn和CUDA已安装并正常运行。图像中提供了旧版本（0.11），并且正在运行CUDA和cudnn，但我需要升级到版本1或更高版本。我有两个Nvidia Titans。

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

pip install tensorflow-gpu

在此之后我只得到输出

没有名为tensorflow的模块

如果我用以下方式查看点子列表：

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-gpu（1.0.1）

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

如果我使用非gpu-support install pip list | grep tensorflow，则上面的代码给出：

设备映射：没有已知设备。

当然这是由于没有支持gpu。总而言之，如何使用简单的pip安装和1.0以上的版本来使用GPU版本的tensorflow？

Answer 1

使用

安装

conda install tensorflow-gpu

解决了所有问题。

Answer 2

使用以下命令激活tensorflow：

>>>source activate tensorflow

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

2 个答案: