如何在多个GPU节点上获取分配给SLURM作业的GPU ID？

Question

当我向具有两个GPU的节点提交带选项--gres = gpu：1的SLURM作业时，如何获得分配给作业的GPU ID？有没有为此目的的环境变量？我使用的GPU都是NVIDIA GPU。 谢谢。

Answer 1

您可以使用环境变量CUDA_VISIBLE_DEVICES获取GPU ID。此变量是分配给作业的GPU ID的逗号分隔列表。