我安装了Tensorflow for GPU使用:pip install tensorflow-gpu
但是当我为Keras尝试相同时:pip install keras-gpu
它给我一个错误:找不到满足要求的版本
希望有人发现并帮助我!谢谢 !
没有任何keras-gpu
包[更新:现在有,见下面的其他答案]; Keras是一些后端的包装器,包括Tensorflow,这些后端可能有不同的版本,例如tensorflow
和tensorflow-gpu
。但这不适用于Keras本身,应该简单地安装
pip install keras
独立于使用的后端(参见PyPi docs)。
此外,既然您已将问题标记为anaconda
,请注意,通常不建议将您的包管理器(即pip
与conda
)混合使用,而您可能最好从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。
除了建议使用Anaconda软件包管理器之外,在上面的答案中添加了正确的答案,但过时了,现在Anaconda Cloud上有一个keras-gpu软件包。
因此,一旦安装了Anaconda,您只需创建一个新环境,您可以在其中安装keras-gpu并执行命令:
conda install -c anaconda keras-gpu
这将安装Keras以及tensorflow和tensorflow-gpu库作为后端。 (也没有必要单独安装CUDA运行时库和cudnn库,因为它们也包含在软件包中 - 在Windows 10上测试并正常工作)。
添加以上两个答案,确保您的TensorFlow / Keras环境使用Python 3.6。截至2019年5月10日,Keras / TensorFlow与Python 3.7的效果不佳。
我尝试使用Keras / TensorFlow和Python 3.7,我最终不得不重新安装Anaconda,因为它有点破坏了我的Anaconda Prompt。