我无法导入tensorflow-gpu

Question

我已经用pip install tensorflow成功安装了tensorflow，并且一切正常。

我也可以使用pip install tensorflow-gpu成功安装tensorflow-gpu，但无法将其导入我的python脚本中：

import tensorflow-gpu File "<stdin>", line 1 import tensorflow-gpu ^ SyntaxError: invalid syntax

我已经安装了CUDA v9.0并运行Windows 10

Answer 1

pypi上的软件包称为tensorflow-gpu，但您只需使用“ tensorflow”将其导入

 import tensorflow as tf

Answer 2

要确保tensorflow软件包正在使用您的GPU，请执行以下操作：

import tensorflow as tf

sess = tf.Session()

请参见控制台上的输出，如果该输出显示了有关创建会话的GPU信息，如下所示。

在信息中注意“ GeForce 940MX”。另外请注意，如果计算能力得分高于3.5，则Tensorflow将仅使用Nvidia GPU。有关here的更多信息。

如果未使用GPU，则不会输出GPU信息，只会显示类似以下内容：