安装Keras for gpu?

时间:2019-02-14 11:12:26

标签: tensorflow keras pip anaconda gpu

我使用以下命令安装了Tensorflow for GPU:pip install tensorflow-gpu 但是,当我对 Keras 进行相同的尝试:pip install keras-gpu时,我却遇到了一个错误:找不到满足要求的版本

希望有人找到并帮助我! 谢谢 !

4 个答案:

答案 0 :(得分:5)

没有没有任何keras-gpu软件包[ UPDATE 现在,请参见下面的其他答案]; Keras是围绕某些后端(包括Tensorflow)的包装,这些后端可能有不同的版本,例如tensorflowtensorflow-gpu。但这不适用于Keras本身,应与

一起简单安装
pip install keras

与使用的后端无关(请参见PyPi docs)。

此外,由于您也已将问题标记为anaconda,因此,通常不建议将软件包管理器(即pipconda混合使用,并且您最好使用{p> 3从Anaconda cloud安装Keras,

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

答案 1 :(得分:4)

在上面的答案中添加了建议使用Anaconda软件包管理器的正确答案,但是由于现在现在已经是 keras-gpu 放在Anaconda Cloud上。

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

conda install -c anaconda keras-gpu

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

答案 2 :(得分:1)

除了上述两个答案外,请确保您的TensorFlow / Keras环境正在使用Python 3.6。截至2019年5月10日,Keras / TensorFlow在Python 3.7上无法很好地工作。

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

答案 3 :(得分:1)

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

“ conda安装tensorflow-gpu cudatoolkit = 9.0 -c anaconda”

类似于keras-gpu