一、问题

安装了tensorflow-gpu后调用keras无法使用gpu进行加速反而使用CPU训练导致程序运行相对缓慢。

二、原因

若我们同时安装了tensorflowtensorflow-gpu,使用keras时会默认调用tensorflow,从而无法使用GPU进行训练。

三、解决方法

同时卸载tensorflow、tensorflow-gpu 和 keras;

然后,再安装tensorflow-gpu 和 keras,即可在使用keras时用gpu进行加速。

 
 
 
Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐