一、问题
在安装了tensorflow-gpu后,调用keras无法使用gpu进行加速,反而使用CPU训练导致程序运行相对缓慢。
二、原因
若我们同时安装了tensorflow和tensorflow-gpu,使用keras时会默认调用tensorflow,从而无法使用GPU进行训练。
三、解决方法
同时卸载tensorflow、tensorflow-gpu 和 keras;
然后,再安装tensorflow-gpu 和 keras,即可在使用keras时用gpu进行加速。
在安装了tensorflow-gpu后,调用keras无法使用gpu进行加速,反而使用CPU训练导致程序运行相对缓慢。
若我们同时安装了tensorflow和tensorflow-gpu,使用keras时会默认调用tensorflow,从而无法使用GPU进行训练。
同时卸载tensorflow、tensorflow-gpu 和 keras;
然后,再安装tensorflow-gpu 和 keras,即可在使用keras时用gpu进行加速。