这篇文章主要介绍了已安装tensorflow-gpu但keras无法使用GPU加速怎么办,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。
创新互联建站-专业网站定制、快速模板网站建设、高性价比繁昌网站开发、企业建站全套包干低至880元,成熟完善的模板库,直接使用。一站式繁昌网站制作公司更省心,省钱,快速模板网站建设找我们,业务覆盖繁昌地区。费用合理售后完善,十年实体公司更值得信赖。问题
我们使用anoconda创建envs环境下的Tensorflow-gpu版的,但是当我们在Pycharm设置里的工程中安装Keras后,发现调用keras无法使用gpu进行加速,且使用的是cpu在运算,这就违背了我们安装Tensorflow-gpu版初衷了。
原因
因为我们同时安装了tensorflow和tensorflow-gpu(在…Anaconda3\envs\fyy_tf\Lib\site-packages中可以找到他们的文件夹),使用keras时会默认调用tensorflow,从而无法使用GPU进行训练。
解决方法
同时卸载tensorflow、tensorflow-gpu 和 keras(可以直接去…Anaconda3\envs\fyy_tf\Lib\site-packages下删除他们的文件夹);
然后,再在Pycharm设置中使用小加号安装tensorflow-gpu 和 keras。
最后就可以使用keras进行gpu加速。
感谢你能够认真阅读完这篇文章,希望小编分享的“已安装tensorflow-gpu但keras无法使用GPU加速怎么办”这篇文章对大家有帮助,同时也希望大家多多支持创新互联成都网站设计公司,关注创新互联成都网站设计公司行业资讯频道,更多相关知识等着你来学习!
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、网站设计器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。