自动选择设备keras tensorflow

时间:2018-06-06 12:48:17

标签: python tensorflow keras

我使用python35,keras 2.1.5& 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

0 个答案:

没有答案