将GPU与Keras结合使用

时间:2018-07-23 10:23:48

标签: python tensorflow cuda keras

自上周五以来,我实际上正面临问题,并且暂时未找到解决方案。

首先,您需要知道我是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 个答案:

答案 0 :(得分:2)

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

假设您正在使用conda,并且希望整理所有这些内容。

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

如果您不是,那么我强烈建议Anaconda处理这些似乎没有压力的问题。