Tensorflow后端的Keras能否被迫随意使用CPU或GPU？

如果要强制Keras使用CPU

方式1

import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"   # see issue #152os.environ["CUDA_VISIBLE_DEVICES"] = ""

在导入Keras / Tensorflow之前。

方式二

运行脚本为

$ CUDA_VISIBLE_DEVICES="" ./your_keras_pre.py

也可以看看

1. https://github.com/keras-team/keras/issues/152
2. https://github.com/fchollet/keras/issues/4613