KerasでGPUを使わない方法
(2019/02/16現在)私の場合は、CUDA10がインストールされているパソコンを使用しています。CUDA10は、現在のバーションのKerasには対応していないので、GPUを使うことができません。
したがって、GPUを選択してしまうと、エラーが吐かれます。。。
勝手にGPUが使われてしまうと困る場合の、解決法として、GPUを使わない方法があります。
プログラム
下のコードを追加することで、GPUを使わずに、CPUだけを使用できるようになります。
import tensorflow as tf
import keras.backend as K
K.tensorflow_backend.set_session(tf.Session(config=tf.ConfigProto(device_count = {'GPU': 0})))
