将 GPU 与 Keras 一起使用答案

【问题标题】：Using GPU with Keras将 GPU 与 Keras 一起使用
【发布时间】：2018-12-30 18:35:05
【问题描述】：

自上周五以来，我实际上面临一个问题，暂时没有找到解决方案。

首先，你要知道我是一个linux初学者，我在实习的时候尝试做一些深度学习，我发现即使我公司有1080 Ti，keras也没有使用它，所以有工作来纠正这个问题。

我正在尝试将 Keras 与 GPU 一起使用。我按照以下步骤安装 TensorFlow：https://www.tensorflow.org/install/install_linux 我还安装了 CUDA、cuDNN。

我在我的机器上发现了一个较旧的 CUDA 安装（版本 7.5）。我安装了 9.2 版而没有卸载 7.5 版。我添加了 PATH 变量，但似乎没有考虑到：[][https://i.stack.imgur.com/B3Pqm.png]

我试图卸载 CUDA 7.5 版但我不知道该怎么做，因为在 usr/local 文件夹中，没有 cuda-7.5 文件夹。

当我在提示中输入 nvidia-smi 时，它可以正常工作。我安装了 tensorflow 和 tensorflow-gpu，但我不工作：[][https://i.stack.imgur.com/78gPd.png]

有人知道如何帮助我吗？我想我的问题的解决方案对于了解 Ubuntu 的人来说并不复杂，而且我觉得我在做一些我不太了解的事情时会失去很多次。

如果有人需要更多信息来帮助我，请随时询问。

谢谢

【问题讨论】：

【解决方案1】：

卸载 tensorflow 并仅安装 tensorflow-gpu。您不应该同时安装两者。如果您使用的是 keras，请安装 keras-gpu。

假设您正在使用 conda，并且想要整理所有这些。做

conda remove keras
conda remove tensorflow*
conda install keras-gpu

如果你不是，那么我强烈推荐Anaconda 来处理这些你似乎没有压力的问题。

【讨论】：

感谢您的回答。我听从了你的指示。我正在使用 pip 并且 pio 不适用于安装 keras-gpu。我尝试使用 conda 并且它有效。但是，当我启动 python 并尝试导入 keras 时，我收到以下消息： ModuleNotFoundError: No module named 'tensorflow.python.framework' 对于我的第二个问题（我的计算机使用 CUDA 7.5 版但我安装了 9.2 版）我通过更改一些环境变量找到了解决方案。我检查了一下，cuDNN 似乎在这里并且可以正常工作（用样本测试）。
您是否先删除了 pip 安装？那将是一个很好的步骤。 'pip uninstall tensorflow' 也是如此，并重复其他可能有冲突的模块。例如 tensorflow-gpu 和 keras。此外，您也有可能为用户和用户安装了东西 su 。所以我也会做'sudo pip uninstall tensorflow'，以防万一。
我用 pip 删除了 tensorflow 和 keras。然后我尝试了一个 pip install --upgrade tensorflow-gpu，然后它工作了！现在我的训练时间是 2 分钟/epoch，而不是 23 分钟。我仍然不明白为什么它会起作用，但没关系。感谢您的帮助！
太棒了。在这种情况下，你能接受这个答案吗？