在Keras中使用GPU

Question

从上周五开始我就面临着一个问题，目前还没有找到解决办法。在

首先，你要知道，我是linux的初学者，我在实习期间试图做一些深入的学习，我发现即使我的公司有一个1080TI，keras也没有使用它，所以要努力纠正这个问题。在

我正在尝试将Keras与GPU一起使用。我通过以下步骤安装Tensorflow:https://www.tensorflow.org/install/install_linux 我也安装CUDA，cuDNN。在

我在我的机器上发现了一个旧版本的CUDA（7.5版）。我安装了9.2版而没有卸载7.5版。我添加了路径变量，但似乎没有考虑到：[][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的人来说并不复杂，我觉得我在做一些我并不真正理解的事情时，会松了一口气。在

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

谢谢你

Answer 1

卸载tensorflow并仅安装tensorflow gpu。您不应该同时安装这两个。如果您正在使用keras，请安装keras gpu。在

假设你和康达一起工作，你想把这一切都收拾好。做

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

如果你没有，那么我强烈建议你用Anaconda来处理这些看起来没有压力的问题。在