マルチGPU環境で、計算に使用するCUDAデバイスを選択する

DRANK

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