我無法導入 tensorflow-gpu

Question

我已經使用pip install tensorflow成功安裝了 tensorflow，並且一切正常。

我也可以使用pip install 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。 更多關於這里。

如果它沒有使用 GPU，那么它不會輸出 GPU 信息，它只會顯示類似這樣的內容：