gpu - Slurm 超额订阅 GPU

Question

有没有办法在 Slurm 上超额订阅 GPU，即运行共享一个 GPU 的多个作业/作业步骤？我们只找到了超额订阅 CPU 和内存的方法，但没有找到 GPU。

我们希望在同一 GPU 上并行运行多个作业步骤，并可选择指定用于每个步骤的 GPU 内存。

Answer 1

最简单的方法是将 GPU 定义为 feature 而不是 gres 因此 Slurm 不会管理 GPU，只需确保该工作需要在提供一个的节点上登陆。