Keras Multi-gpu：指定显式GPU ID

Question

通过查看tensorflow GitHub存储库中的文件keras/utils/multi_gpu_utils.py，我可以看到，假设您指定要使用x GPU，它将自动从{{1}中分配GPU ID。 }，即range(x)。

我需要使用GPU 0, 1, 2, ..., x - 1，因为前4个GPU已经在执行其他任务。有没有办法指定它？

Answer 1

在python中，您可以使用

import os
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"

或者在启动python脚本之前在bash中设置CUDA_VISIBLE_DEVICES=0,1

您也可以参考我的答案here使该过程自动化。