已安装tensorflow-gpu，但keras无法使用GPU加速 解决方法
·
一、问题
在安装了tensorflow-gpu后，调用keras无法使用gpu进行加速，反而使用CPU训练导致程序运行相对缓慢。
二、原因
若我们同时安装了tensorflow和tensorflow-gpu，使用keras时会默认调用tensorflow，从而无法使用GPU进行训练。
三、解决方法
同时卸载tensorflow、tensorflow-gpu 和 keras;
然后，再安装tensorflow-gpu 和 keras，即可在使用keras时用gpu进行加速。
新一代开源开发者平台 GitCode，通过集成代码托管服务、代码仓库以及可信赖的开源组件库，让开发者可以在云端进行代码托管和开发。旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。
更多推荐
3
0- 0
已为社区贡献5条内容
所有评论(0)