2019年4月22日
Keras强制使用CPU
摘要： 最近使用GPU来跑Keras模型速度很快，但是预测的时候加载的非常慢，估计是使用GPU的问题 GPU做并行运算效果优于CPU，但是在预测的时候不需要并行运算，如果再使用GPU的话导致加载时间太长 因此训练模型使用GPU，预测使用CPU 在导入Keras之前加入以下两行代码： 既可以使用CPU来预测模 阅读全文
posted @ 2019-04-22 11:25 Freddy520 阅读(5240) 评论(0) 推荐(0) 编辑