複数のgpu が接続されているときの環境変数の設定

nvidia-smi で複数のgpuが存在するとき

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

['/device:GPU:0']

nvidia-smi