为gpu安装Keras?

问题描述 投票:0回答:3

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

希望有人发现并帮助我!谢谢 !

tensorflow keras pip anaconda gpu
3个回答
5
投票

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

pip install keras

独立于使用的后端(参见PyPi docs)。

此外,既然您已将问题标记为anaconda,请注意,通常不建议将您的包管理器(即pipconda)混合使用,而您可能最好从Anaconda cloud安装Keras

conda install -c conda-forge keras 

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

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

conda install -c anaconda tensorflow-gpu 

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


5
投票

除了建议使用Anaconda软件包管理器之外,在上面的答案中添加了正确的答案,但过时了,现在Anaconda Cloud上有一个keras-gpu软件包。

因此,一旦安装了Anaconda,您只需创建一个新环境,您可以在其中安装keras-gpu并执行命令:

conda install -c anaconda keras-gpu

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


3
投票

添加以上两个答案,确保您的TensorFlow / Keras环境使用Python 3.6。截至2019年5月10日,Keras / TensorFlow与Python 3.7的效果不佳。

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

© www.soinside.com 2019 - 2024. All rights reserved.