【问题标题】:How to install Keras with gpu support?如何安装支持 gpu 的 Keras?
【发布时间】:2019-07-08 09:39:57
【问题描述】:

我为 GPU 安装了 Tensorflow:pip install tensorflow-gpu 但是当我为 Keras pip install keras-gpu 尝试相同的操作时,它给我一个错误:找不到满足要求的版本

【问题讨论】:

  • 请说明使用的 Python 版本。另请注意,keras-gpu 是 conda 安装,而不是 pip 安装。

标签: tensorflow keras pip anaconda gpu


【解决方案1】:

要安装具有特定 cuda 版本 9.0 的 tensorflow-gpu,请使用:

“conda install tensorflow-gpu cudatoolkit==9.0 -c anaconda”

keras-gpu 类似

【讨论】:

    【解决方案2】:

    除了以上两个答案,请确保您的 TensorFlow/Keras 环境使用 Python 3.6。截至 2019 年 5 月 10 日,Keras/TensorFlow 不能很好地与 Python 3.7 配合使用。

    我尝试在 Python 3.7 中使用 Keras/TensorFlow,但最终不得不重新安装 Anaconda,因为它破坏了我的 Anaconda Prompt。

    【讨论】:

      【解决方案3】:

      没有任何keras-gpu 包[更新现在有,请参阅下面的其他答案]; Keras 是一些后端的封装,包括 Tensorflow,这些后端可能有不同的版本,比如tensorflowtensorflow-gpu。但这不适用于 Keras 本身,它应该简单地安装

      pip install keras
      

      独立于使用的任何后端(请参阅PyPi docs)。

      此外,由于您也将问题标记为anaconda,请注意通常不建议将您的包管理器(即pipconda)混合使用,您最好安装Keras 来自Anaconda cloud

      conda install -c conda-forge keras 
      

      最后,您可能还想知道最新版本的 Tensorflow 包含 Keras 作为子包,因此您无需任何额外安装即可使用它;见https://www.tensorflow.org/guide/keras

      要从Anaconda cloud 安装tensorflow-gpu,您应该使用

      conda install -c anaconda tensorflow-gpu 
      

      在安装 Keras 之前。确保您在不同的虚拟环境中执行此操作,或者在卸载其他版本(即pip-安装的版本)之后执行此操作,否则有reported problems

      【讨论】:

      • 所以改为 pip install tensorflow-gpu .. 我可以在 anaconda 中安装什么
      • @SaiKrishnadas 查看更新的答案(无论如何,这是一个简单的谷歌搜索......);如果答案有帮助,请接受 - 谢谢
      • Still.. 系统信息 * Windows 7 * pip install tensorflow-gpu * Python版本: 3.6.6 * 使用 virtualenv 安装?点子? conda?:pip * CUDA/cuDNN 版本:CUDA 8.0 cuDNN 7.1 * GPU 型号和内存:型号:Quadro 4000,内存:18GB 已安装使用 pip install tensorflow-gpu (没有错误,完美安装)在导入 tensorflow 时,出现错误:Failed to load native tensorflow .ImportError: DLL load failed: The specified module could not be found跨度>
      • 您最初的问题是关于keras-gpu,已在答案中解决。如果您对不同的软件包有不同的问题,非常欢迎您在接受这个问题后提出一个新问题。
      【解决方案4】:

      添加到上面的答案中,这是推荐使用 Anaconda 包管理器的正确答案,但已经过时了 现在有一个 keras-gpuAnaconda Cloud上打包。

      所以一旦你安装了 Anaconda,你只需要创建一个要安装 keras-gpu 的新环境并执行命令:

      conda install -c anaconda keras-gpu

      这将安装 Keras 以及 tensorflow 和 tensorflow-gpu 库作为后端。 (也无需单独安装 CUDA 运行时和 cudnn 库,因为它们也包含在包中 - 在 Windows 10 上测试并正常工作)。

      【讨论】:

      • 在这种情况下,keras 使用tensorflow CPU 版本作为后端,根本不使用 GPU。知道如何解决这个问题吗?
      • 先安装 tensorflow-gpu 然后安装 keras-gpu 成功了! stackoverflow.com/a/69498829/3337089
      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2021-12-11
      • 1970-01-01
      • 1970-01-01
      • 2019-03-14
      • 2020-09-02
      相关资源
      最近更新 更多