Keras强制使用CPU

最近使用GPU来跑Keras模型速度很快，可是预测的时候加载的很是慢，估计是使用GPU的问题html GPU作并行运算效果优于CPU，可是在预测的时候不须要并行运算，若是再使用GPU的话致使加载时间太长post 所以训练模型使用GPU，预测使用CPUspa   在导入Keras以前加入如下两行代码：code import os os.environ["CUDA_VISIBLE_DEVICES"]

>>阅读原文<<