自动选择设备keras tensorflow

Question

我使用python35，keras 2.1.5＆amp; Tensorflow GPU 1.8.0

我同时训练多个模型（以及同事也在4 gpu的集群上）。

我已尝试使用以下代码选择GPU：

from keras import backend as K
import os

os.environ["CUDA_VISIBLE_DEVICES"]="0" #or 1, etc

K.tensorflow_backend._get_available_gpus()

可悲的是，使用这种方法我需要手动选择一个可用的GPU（使用nvidia-smi，选择一个，并在nohup启动和nvidia命令之间祈祷没人开始训练）

有没有办法自动选择GPU？

另外，有没有办法阻止其他用户/程序访问这个GPU？

- 编辑： -

对于第二个问题，there is also another duplicate explaining how to allocate only a fraction of the GPU memory