Keras无法调用tensorflow-gpu的解决方案
·
今天用keras训练时发现内存占用率出奇的高，而且显存占用率出奇的低，原来keras没有用gpu训练。
通过
pip list看到同时安装了tensorflow和tensorflow-gpu，keras默认调用了tensorflow
解决办法
同时卸载tensorflow，tensorflow-gpu，keras
再安装tensorflow-gpu和keras就可以解决
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念，把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起，为开发者提供从开发、训练到部署的一站式体验。
更多推荐
3
0- 0
已为社区贡献5条内容
所有评论(0)