使用python提交作业

时间:2013-10-17 16:17:02

标签: python jobs

我正在尝试使用python脚本在我们机构的集群中提交作业。

 compile_cmd = 'ifort -openmp ran_numbers.f90 ' + fname \
                  + ' ompscmf.f90 -o scmf.o'
 subprocess.Popen(compile_cmd, shell=True)

 Popen('qsub launcher',shell=True)

问题是,此时系统悬而未决。上述脚本中有任何明显错误吗?代码中提到的所有文件都可以在该目录中找到(我已经交叉检查过了)。 qsub 是用于向我们的群集提交作业的命令。 fname 是我在此过程中创建的文件的名称。

2 个答案:

答案 0 :(得分:5)

我有一个脚本,我曾经使用qsub将多个作业提交到我们的集群。 qsub通常以

的形式提交作业
qsub [qsub options] job

在我的工作中,job通常是一个bash(.sh)或python脚本(.py),它实际上调用要在每个节点上运行的程序或代码。如果我想提交一个名为“test_job.sh”的作业,最长时间,我会做

qsub -l walltime=72:00:00 test_job.sh

这相当于以下python代码

from subprocess import call

qsub_call = "qsub -l walltime=72:00:00 %s"
call(qsub_call % "test_job.sh", shell=True)

或者,如果您的bash脚本看起来像

,该怎么办?
#!/bin/bash

filename="your_filename_here"
ifort -openmp ran_numbers.f90 $filename ompscmf.f90 -o scmf.o

然后通过qsub job.sh

提交了此内容

编辑:老实说,最佳的作业排队方案因群集而异。简化作业提交脚本的一种简单方法是找出每个节点可用的CPU数量。一些较新的排队系统允许您提交许多单个CPU作业,他们将在尽可能少的节点上一起提交这些作业;然而,一些较旧的集群不会这样做并且提交许多个人工作是不受欢迎的。

假设群集中的每个节点都有8个CPU。你可以写下像

这样的脚本
#!/bin/bash
#PBS -l nodes=1;ppn=8

for ((i=0; i<8; i++))
do
    ./myjob.sh filename_${i} &
done
wait

这样做的目的是一次在一个节点上提交8个作业(&表示在后台执行)并等到所有8个作业完成。对于每个节点具有多个CPU的集群,这可能是最佳的(例如,我使用的一个集群每个节点有48个CPU)。

或者,如果提交许多单核作业是最佳的,并且上面的提交代码不起作用,您可以使用python生成bash脚本以传递给qsub。

#!/usr/bin/env python
import os
from subprocess import call

bash_lines = ['#!/bin/bash\n', '#PBS -l nodes=1;ppn=1\n']
bash_name = 'myjob_%i.sh'
job_call = 'ifort -openmp ran_numbers.f90 %s ompscmf.f90 -o scmf.o &\n'
qsub_call = 'qsub myjob_%i.sh'

filenames = [os.path.join(root, f) for root, _, files in os.walk(directory)
                                   for f in files if f.endswith('.txt')]
for i, filename in enumerate(filenames):
    with open(bash_name%i, 'w') as bash_file:
        bash_file.writelines(bash_lines + [job_call%filename, 'wait\n'])
    call(qsub_call%i, shell=True)

答案 1 :(得分:0)

你有任何错误吗?因为你似乎错过了第二个Popen的“ subprocess。”。