本文介绍了在Windows10上已安装tensorflow-gpu的前提下，如何避免conda自动安装CPU版tensorflow，正确使用pip安装keras-gpu的步骤。作者通过亲身经历警告读者不要使用conda安装keras，以免得到CPU版本。安装成功后，GPU加速的keras在GTX 1060上运行速度远超CPU版。
【注】本教程是在已经安装tensorflow-gpu1.7.0基础上，安装keras-gpu版本的一些经验！
第一次安装keras的时候，也不清楚什么cpu版本的还有gpu版本的，就百度按照别人的方法直接使用命令conda install keras进行安装，结果。。。它弹出来了提示说还需要附带安装theano以及tensorflow1.1.0版本以及其他几个库，什么也不懂，就同意了，结果在跑keras程序的时候，查看自己的显卡状态，使用率始终为0%，也就是说安装的这个keras根本没有使用gpu加速，而是使用的它自带的那个tensorflow1.1.0版本，这应该是默认使用cpu运算的，整个跑程序的过程很慢，就想着卸载keras，重新进行安装，直接使用命令：
pip install keras
注：这里使用pip安装而不是使用conda，原因是使用
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
