GPUが複数刺さった計算機でKeras(Tensorflow backend)を使おうとしたところ、 Tensorflowが全GPUのVRAMを全て確保しようとするために、他の計算が回らなくなって困った。 GPU1枚だけを専有するように指定する方法でハマったのでメモしておく。 ソリューション 環境変数のCUDA_VISIBLE_DEVICESで指定できる。 kerasやtensorflowをimportする前に宣言すること import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = ""