python - 将 GPU 与 Keras 一起使用

Question

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

首先，你需要知道我是一个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 的人来说并不复杂，而且我觉得我在做一些我不太了解的事情时会失去很多次。

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

谢谢

标签： pythontensorflowcudakeras

Answer 1

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

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

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

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