欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

已安装tensorflow-gpu,但keras无法使用GPU加速的解决

时间:2020-11-01 14:41:07|栏目:Python代码|点击:

问题

我们使用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加速。

上一篇:Python去除字符串前后空格的几种方法

栏    目:Python代码

下一篇:Python3实现将本地JSON大数据文件写入MySQL数据库的方法

本文标题:已安装tensorflow-gpu,但keras无法使用GPU加速的解决

本文地址:http://www.codeinn.net/misctech/18172.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有