我使用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