最近使用GPU来跑Keras模型速度很快，但是预测的时候加载的非常慢，估计是使用GPU的问题
GPU做并行运算效果优于CPU，但是在预测的时候不需要并行运算，如果再使用GPU的话导致加载时间太长
因此训练模型使用GPU，预测使用CPU
在导入Keras之前加入以下两行代码：
import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
既可以使用CPU来预测模型
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。