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

Question

当我使用选项 --gres=gpu:1 向具有两个 GPU 的节点提交 SLURM 作业时，如何获取为该作业分配的 GPU ID？是否有用于此目的的环境变量？我使用的 GPU 都是 nvidia GPU。谢谢。

Answer 1

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