我用的一个服务器上装了一个集群管理工具(SGE, Sun Grid Engine), 用于从登陆节点上向计算节点进行任务投递。一开始,不太会用,但是经过一段时间的摸索学习后,终于能顺手的用起来了。
在使用SGE之前,你得先了解SGE到底做了什么事情. SGE或者其他集群管理工作做的事情就是将用户投递的任务进行排队,然后将任务交给能够运行的结算节点执行,工作流程可以分为四步:
- 接受用户投放的任务
- 在任务运行以前,将任务放到一个存储区域
- 发送任务到一个执行设备,并监控任务的运行
- 运行结束写回结果并记录运行日志
SGE的常用命令
任务投递
SGE中投递任务所用到的命令是qsub
. 最简单的用法是下面这种,即,将要执行的命令通过标准输入的方式传递给qsub
echo "ls -l " | qsub
投递之后可以用qstat
查看自己投递的任务的运行情况,如下图
第一列是任务编号, 第二列是优先级,第三列是任务名字,在参数里没有特别说明的情况下,SGE会用任务的来源进行命令,STDIN表示来自于标准输入,第四列是用户名,第五列是运行状态("r"表示运行中), 第六列表示任务投递和开始时间,第七列是任务投递的节点,第8列则是要申请的线程数。在执行完成后会在家目录下生成"STDIN.e7883"和"STDIN.o7883", 其中7883就是任务编号, 前者存放标准错误输出, 后者存放标准输出, 因此"cat STDIN.o7883"的内容就是ls -l
的内容。
另一种方法是先写一个脚本然后投递,比如先编辑一个文件"ls.sh", 内容如下,然后用"qsub ls.sh"投递任务。
ls -l
跟之前一样,最后在家目录下产生了"ls.sh.exxxx"和"ls.sh.exxxx"两个文件
当然实际时肯定没有那么简单,我们需要增加各种参数来调整qsub
的行为,用qsub -help
可以看完整的参数,但是常用的为如下几个
- -q xxx : 指定要投递到的队列,如果不指定的话,SGE会在用户可使用的队列中选择一个满足要求的队列。
- -V: 将当前的环境变量传递到执行命令的节点中
- cwd: 在当前目录下执行任务, sge的日志会输出到当前路径。 不增加该指令,所有投递的任务都会在家目录下执行
- -l resource=value: 请求资源数, 例如
-l vf=25G -l h=node1
就是任务的预估内存要25G(内存估计的值应稍微大于真实的内存,内存预估偏小可能会导致节点跑挂), 申请在node1上运行 - -S /bin/bash: 表示在bash环境下执行命令。默认tcsh.
- -pe openmpi 4: 表示使用openmpi进行并行运算,且申请的线程是4,
- -N 任务名: 手动执行任务的名字
- -j y|n :是否将标准输入和标准输入合并成一个文件
- -sync y|n: 是否等待任务结束,返回退出码
- -o path: 指定标准输出的文件夹
那么接下来就可以添加这些参数运行一些命令了,例如在命令行里投递一个比对任务
echo "bowtie2 -p 8 -x index/ref -1 data/A_1.fq -2 data/A_2.fq | samtools sort > A.bam" | qsub -V -cwd -l vf=25G -S /bin/bash -pe openmpi 8 -N A.bt2
这些参数除了在外部设置外,还可以在shell脚本里设置,如下
#!/bin/bash
#$ -S /bin/bash
#$ -V
#$ -cwd
#$ -l vf=25G
#$ -pe openmpi 8
#$ -N a.bt2
bowtie2 -p 8 -x index/ref -1 data/A_1.fq -2 data/A_2.fq | samtools sort > A.bam
查询任务
除了任务投递外,查询任务也是一个非常常用的命令,除了刚才直接用qstat
查看,还有如下参数比较好用
qstat -f # 查看用户任务
qstat -j jobId # 按任务id查看
qstat -explain a|c|A|E -j jobID # 查看任务任务并给出解释
qstat -u user # 按用户查看
任务状态:
- qw: 表示等待状态
- hqw: 任务挂起等待中,待依赖的任务完成后执行
- Eqw: 投递任务出错
- r: 表示任务正在运行
- s: 暂时挂起
- dr: 节点挂了之后,删除任务就会出现这个状态,只有节点重启之后,任务才会消失
删除任务
任务删除也比较重要,毕竟偶尔会出现任务投递出错的情况
qdel -j 1111 删除任务号为1111的任务
其他命令
- qrsh:与qsub相比,是交互式的投递任务,注意参数:
-now yes|no
默认设置为yes- 若设置为yes,立即调度作业,如果没有可用资源,则拒绝作业,任务投递失败,任务状态为Eqw。
- 若设置为no,调度时如果没有可用资源,则将作业排入队列,等待调度。
- 例子: qrsh -l vf=*G -q all.q -now no -w n *sh
- qacct 从集群日志中抽取任意账户信息
- qalter 更改已提交但正处于暂挂状态的作业的属性
- qconf 为集群和队列配置提供用户界面
-
qconf -spl
查看可用并行环境
-
- qhold 阻止已提交作业的执行
- qhost 显示SGE执行主机(即各个计算节点)的状态信息
-
qhost -j
按照节点显示任务 -
qhost -F
展示每个节点的资源
-
- qlogin 启动telnet或类似的登录会话。
案例:一个投递比对任务的简单脚本
#!/bin/bash
set -e
set -u
set -o pipefail
threads=8
index=index/hg19
FQ_DIR="analysis/0-raw-data"
ALIGN_DIR="analysis/2-read-align"
LOG_DIR="analysis/log"
TMP_DIR="analysis/tmp"
mkdir -p ${ALIGN_DIR}
mkdir -p ${LOG_DIR}
mkdir -p ${TMP_DIR}
tail -n+2 download_table.txt | cut -f 6 | while read id;
do
echo "
bowtie2 --very-sensitive-local --mm -p $threads -x $index -U ${FQ_DIR}/$id.fastq.gz 2> ${LOG_DIR}/$id.bt2.log | \
samtools sort -@ 2 -m 1G -T ${TMP_DIR}/${id} -o ${ALIGN_DIR}/${id}.sort.bam" | qsub -V -cwd -pe openmpi $threads -N ${id}_bt2 -q all.q -S /bin/bash
done