KerasでGPUを使わない方法

ディープラーニング




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})))

参考

Jupyter上で使用するGPUを指定する最適解

タイトルとURLをコピーしました