[解決済み] Tensorflowバックエンドを持つKerasはCPUとGPUを任意に使い分けられるか？

質問

私はTensorflowバックエンドとCUDAと一緒にKerasをインストールしています。 時々オンデマンドでKerasがCPUを使用するように強制したいです。 これは、仮想環境に別のCPU専用のTensorflowをインストールするといったことをせずに行うことができますか？ もしそうなら、どのように？ バックエンドがTheanoであれば、フラグを設定することができますが、私はKerasを介してアクセス可能なTensorflowフラグを聞いたことがありません。

どのように解決するのですか？

Kerasに強制的にCPUを使わせたい場合

方法1

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

Keras / Tensorflowがインポートされる前に。

ウェイ2

としてスクリプトを実行します。

$ CUDA_VISIBLE_DEVICES="" ./your_keras_code.py

参照

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