我无法导入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

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

import tensorflow as tf

sess = tf.Session()

查看控制台上的输出，如果它显示了有关创建会话的GPU信息，如下所示。

请注意信息中的"GeForce 940MX“。另请注意，如果计算能力得分高于3.5时，Tensorflow将使用Nvidia GPU only。更多关于here的信息。

​

如果它不使用GPU，那么它不会输出GPU信息，它只会显示类似如下的内容：

​

​

Answer 2

pypi上的包被称为tensorflow-gpu，但您只需使用"tensorflow“导入它

 import tensorflow as tf