複数GPU所持しているPCでKeras内に使用GPUの指定方法

使用方法

import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto(
    gpu_options=tf.GPUOptions(
        visible_device_list="2", # specify GPU number
        allow_growth=True
    )
)
set_session(tf.Session(config=config))

(https://github.com/fchollet/keras/issues/1538)

基本的にはバックエンドとして動作するTensorflowの設定を変更します．keras.backend.tensorflow_backend.set_sessionを用います． 注意点としては，コードの内部でtf.Sessionを取得しているようなケースでは，先頭に上記を記述しても意味がありません（とくに自分で書いていないコードを走らせる場合）．

参考

qiita.com