我无法导入tensorflow gpu

Question

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

我也可以成功地用pip Instentorflow-gpu安装tenstorflow-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”。还请注意，Tensorflow将仅在计算能力得分高于3.5时使用Nvidia GPU。更多关于这方面的信息。

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

Answer 2

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

 import tensorflow as tf