自上周五以来,我实际上正面临问题,并且暂时未找到解决方案。
首先,您需要知道我是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的人来说并没有那么复杂,我觉得我失去了很多时间去做一些我不太了解的事情。
如果有人需要更多信息以帮助我,请随时询问。
谢谢
答案 0 :(得分:2)
卸载tensorflow并仅安装tensorflow-gpu。您不应同时安装两者。如果使用的是keras,请安装keras-gpu。
假设您正在使用conda,并且希望整理所有这些内容。
conda remove keras
conda remove tensorflow*
conda install keras-gpu
如果您不是,那么我强烈建议Anaconda处理这些似乎没有压力的问题。