diff --git a/meshroom/submitters/slurmSubmitter.py b/meshroom/submitters/slurmSubmitter.py index 86ebd1a9..6fe53f77 100644 --- a/meshroom/submitters/slurmSubmitter.py +++ b/meshroom/submitters/slurmSubmitter.py @@ -39,7 +39,6 @@ class SlurmSubmitter(BaseSubmitter): print(node.name) - command = [ "sbatch", "--parsable", @@ -52,6 +51,10 @@ class SlurmSubmitter(BaseSubmitter): dependenciesString = "%s:%d" % (dependenciesString, dependency.pid) command.append(dependenciesString) + if node.nodeDesc.gpu.name != "NONE": + command.append("--partition=gpu") + command.append("--gres=gpu:1") + countCores=24 if node.isParallelized: blockSize, fullSize, nbBlocks = node.nodeDesc.parallelization.getSizes(node)