我使用以下命令安装了Tensorflow for GPU:pip install tensorflow-gpu
但是,当我对 Keras 进行相同的尝试:pip install keras-gpu
时,我却遇到了一个错误:找不到满足要求的版本
希望有人找到并帮助我! 谢谢 !
答案 0 :(得分:5)
没有没有任何keras-gpu
软件包[ UPDATE :现在,请参见下面的其他答案]; Keras是围绕某些后端(包括Tensorflow)的包装,这些后端可能有不同的版本,例如tensorflow
和tensorflow-gpu
。但这不适用于Keras本身,应与
pip install keras
与使用的后端无关(请参见PyPi docs)。
此外,由于您也已将问题标记为anaconda
,因此,通常不建议将软件包管理器(即pip
与conda
混合使用,并且您最好使用{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