已安装tensorflow-gpu,但keras无法使用GPU加速 解决方法
·
一、问题
在安装了tensorflow-gpu后,调用keras无法使用gpu进行加速,反而使用CPU训练导致程序运行相对缓慢。
二、原因
若我们同时安装了tensorflow和tensorflow-gpu,使用keras时会默认调用tensorflow,从而无法使用GPU进行训练。
三、解决方法
同时卸载tensorflow、tensorflow-gpu 和 keras;
然后,再安装tensorflow-gpu 和 keras,即可在使用keras时用gpu进行加速。
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐



所有评论(0)