使用Jenkins和nvidia-docker排队多个GPU

Question

我在拥有4个GPU的计算机上运行Jenkins并使用nvidia-docker运行Jenkins作业以使用GPU。我可以将NVIDIA_VISIBLE_DEVICES属性传递给nvidia-docker，让我指定容器中可以访问哪些GPU。我想要做的是使用Jenkins来管理GPU资源并相应地排队作业，即制作参数化作业，询问用户作业需要多少GPU，并且可以在作业可用时排队并启动作业。理想情况下，我希望在多个从节点上执行此操作，例如每个节点有4个GPU。

这看起来有可能吗？

Answer 1

lock步骤应该会让你更进一步。它提供了几个选项来创建和使用/锁定对几个构建之间共享的资源的访问。

请注意，该作业将在执行期间声明并被阻止（因此请勿将lock置于node分配中。