Keras强制使用CPU
Posted yqs-0705
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Keras强制使用CPU相关的知识，希望对你有一定的参考价值。
最近使用GPU来跑Keras模型速度很快，但是预测的时候加载的非常慢，估计是使用GPU的问题
GPU做并行运算效果优于CPU，但是在预测的时候不需要并行运算，如果再使用GPU的话导致加载时间太长
因此训练模型使用GPU，预测使用CPU
在导入Keras之前加入以下两行代码：
import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
既可以使用CPU来预测模型
以上是关于Keras强制使用CPU的主要内容，如果未能解决你的问题，请参考以下文章
是否有 R 命令使 Keras Tensorflow-GPU 在 CPU 上运行？
如何强制 keras 使用 tensorflow GPU 后端
在 keras 中使用带有 LSTM 的 CNN 时，池化层是强制性的吗？
训练某些网络时，GPU 上的 Keras（Tensorflow 后端）比 CPU 上慢