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

- Seppukki

---

可能是 https://dev59.com/safja4cB1Zd3GeqP2uj8 的重复问题。 - amanb

如果您已正确安装了 cuda 和 cudnn，则无需导入 tensorflow-gpu，您将能够从 tensorflow 访问 gpu。 - Dinari

但是我怎么确保tensorflow使用我的GPU呢？你的意思是如果我在脚本中导入tensorflow，它会自动使用我的GPU吗？ - Seppukki

要查看 TensorFlow 知道哪些 GPU，除非明确定义，否则它将使用 GPU（如果可能）。 - Dinari

当我执行 device_lib.list_local_devices() 时，我的GPU没有出现，可能是因为我在虚拟环境中，它找不到我的GPU吗？ - Seppukki

Answer 1

在pypi上的软件包名为tensorflow-gpu，但是你只需要用"tensorflow"导入它。

 import tensorflow as tf

- MJK

---

您可以首先检查您的GPU是否正在使用。如果您的显卡来自NVidia，则Tensorflow将使用您的GPU。如果您有一张AMD卡，您需要检查您的卡是否受到“PlaidML”的支持。如果是，请安装“PlaidML”，告诉TF使用它，然后开始吧。检查GPU使用情况Windows 10：https://www.windowscentral.com/how-track-gpu-performance-data-windows-10PlaidML：https://github.com/plaidml/plaidml - pookie

它同时具有GPU和CPU实现。但是，当操作分配给设备时，GPU设备将优先考虑。要找出使用的设备，您可以像这样启用日志设备放置 - sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))。 - MJK

我的图形处理器来自NVidia，但它没有使用我的GPU（我有一张GTX1050）。在“sess”中存储的任何内容都是>>> sess <tensorflow.python.client.session.Session object at 0x000001E2587F8C18>。 - Seppukki

Answer 2

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

import tensorflow as tf

sess = tf.Session()

请在控制台上查看输出，如果显示创建会话时的GPU信息如下所示，请注意信息中的“GeForce 940MX”。还要注意，Tensorflow仅在计算能力得分高于3.5时才使用Nvidia GPU。有关更多信息，请参见此处。

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

- Rahul Bharadwaj

---

不，这意味着您正在使用 GPU 版本。我建议通过执行“pip uninstall tensorflow”卸载 tensorflow，然后再次尝试。 - Rahul Bharadwaj

等一下...我应该删除和重新安装哪一个？我只会把它们都删除然后再安装正常的 TensorFlow。 - Seppukki

执行以下命令：pip uninstall tensorflow，我认为你同时安装了CPU和GPU版本，因此出现了混淆。如果你执行 pip uninstall tensorflow-gpu，它将卸载tensorflow GPU。如果你执行 pip uninstall tensorflow，它将卸载tensorflow CPU。 - Rahul Bharadwaj

是的，现在我已经尝试过普通的了，但什么都不起作用，现在我正在尝试使用GPU版的。 - Seppukki

嗯，这很奇怪。你能试试这个：https://dev59.com/PloT5IYBdhLWcg3w8S-2#50538927 - Rahul Bharadwaj