SLURM会根据GRES自动限制内存/ CPU使用率

问题描述 投票:2回答:1

鉴于单个节点具有多个GPU,是否有一种方法可以根据请求的GPU数量自动限制CPU和内存使用?

特别是,如果用户作业脚本请求2个GPU,则该作业应自动限制为2*BaseMEM2*BaseCPU,其中BaseMEM = TotalMEM/numGPUsBaseCPU=numCPUs/numGPUs,将在每个节点的基础上定义。

是否可以通过这种方式配置SLURM?如果不是,是否可以“虚拟地”将一台多GPU的计算机拆分为多个具有适当CPU和MEM计数的节点?

slurm
1个回答
0
投票

在命令行上

--cpus-per-gpu $BaseCPU --mem-per-gpu $BaseMEM

在slurm.conf

DefMemPerGPU=1234
DefCpuPerGPU=1

由于无法在slurm.conf中使用变量,因此您需要编写一个小的bash命令来计算$ BaseCPU和$ BaseMEM

© www.soinside.com 2019 - 2024. All rights reserved.