作业系统

Slurm系统简介

在公共集群中使用SLURM作业调度系统进行任务的调度和管理。SLURM (Simple Linux Utility for Resource Management)是一种可用于大型计算节点集群的高度可伸缩和容错的集群管理器和作业调度系统,被世界范围内的超级计算机和计算集群广泛采用。

Slurm常用命令

sinfo

查看节点与分区状态

squeue

查看队列状态

scancel

取消作业

sacct

查看历史作业信息

salloc

分配资源

sbatch

提交批处理作业

scontrol

系统控制

srun

执行作业

日常使用超算资源只需掌握简单的几条命令即可,具体详细的配置请参考SLURM官方文档

查询状态

sinfo:查看节点与分区状态

$ sinfo
PARTITION AVAIL  TIMELIMIT  NODES  STATE NODELIST
cpu*         up   infinite      1  down* n144
cpu*         up   infinite    168  alloc n[3-143,145-171]

关键词

含义

PARTITION

分区名,对节点的逻辑分组。不同的分区会设置不同权限、资源限制等。

AVAIL

可用状态:up 可用;down 不可用

TIMELIMIT

该分区的作业最大运行时长限制, 30:00 表示30分钟,如果是2-00:00:00表示2天,如果是infinite表示不限时间

NODES

节点数量

STATE

状态:drain: 排空状态,表示该类节点不再分配到其他;idle: 空闲状态;alloc: 被分配状态; mix:部分被占用,但是仍有可用资源; down停机

NODELIST

节点列表

image0

squeue:查看队列状态

$ squeue
JOBID PARTITION     NAME     USER ST       TIME  NODES NODELIST(REASON)
8628       cpu vasp_cpu  yangx17 PD       0:00      2 (QOSMaxNodePerUserLimit)
8629       cpu vasp_cpu  yangx17 PD       0:00      2 (QOSMaxNodePerUserLimit)
8630       cpu vasp_cpu  yangx17 PD       0:00      2 (QOSMaxNodePerUserLimit)
8636       cpu vasp_cpu  mab2019 PD       0:00      4 (Resources)
8637       cpu     vasp lizhenhu PD       0:00      1 (Priority)
5042       cpu HICE_WAC    xum17  R 16-22:28:14      4 n[114-117]
5044       cpu LICE_WAC    xum17  R 16-22:21:58      4 n[29,41-43]
5519       cpu        c zhaosy16  R 14-22:00:21      5 n[93-95,165-166]
5783       cpu        c liangt20  R 13-20:54:50      5 n[30-32,156-157]

关键词

含义

JOBID

作业的id号,每个成功提交的任务都会有唯一的id

PARTITION

分区名

NAME

作业名称,默认为提交脚本的名称

USER

用户名,提交该作业的用户名

ST

作业状态:PD排队;R运行;S挂起;CG正在退出

TIME

作业运行时间

NODES

作业占节点数

NODELIST(REASON)

作业所占节点列表,如果是排队状态的任务,则会给出排队原因

image1

scancel:取消作业

image2

Slurm作业提交

系统页面提交

系统支持直接在页面提交作业。

点击上方“集群”按钮,选择“提交作业”。

submit job web

选择需要使用的集群和作业模板,填写作业名称,在脚本编辑器里填入作业脚本,点击右上方的“提交作业”按钮。

submit job

提交作业后,可以在“作业”页面查看是否提交成功。

job status

如果提交的作业有输出,需要下载,等待作业运行完成后,点击作业名称,进入作业详情。

enter job info

在“作业输出”页面,点击“下载输出日志”。

download job output

终端提交

Slurm作业通常分为交互式和批量式两种。交互式作业通常用于代码编译、脚本调试、交互式计算等工作。长期后台计算的任务通常以作业脚本的方式进行批量提交。

交互作业

集群的登录节点设置有资源限制,请勿在登录节点进行大量计算。

集群的计算节点默认不允许用户直接登录,对需要交互式处理的程序,在登录到集群后,使用salloc命令分配节点,然后再ssh到分配的节点上进行处理:

image3

计算完成后,使用exit命令推出节点,注意需要exit两次,第一次exit是从计算节点退出到登录节点,第二次exit是释放所申请的资源。

image4

批量作业

可以通过将程序执行命令放入作业提交脚本,并通过sbatch命令作业提交的方式在集群中进行计算。

一个简单的脚本示例如下:

 1#! /bin/bash
 2### 表示这是一个bash脚本
 3
 4#SBATCH --job-name=JOBNAME
 5### 设置该作业的作业名
 6
 7#SBATCH --nodes=2
 8### 指定该作业需要2个节点数
 9
10#SBATCH --ntasks-per-node=40
11### 每个节点所运行的进程数为40
12
13#SBATCH --time=2:00:00
14### 作业最大的运行时间,超过时间后作业资源会被SLURM回收
15
16#SBATCH --comment project_name
17### 指定从哪个项目扣费。如果没有这条参数,则从个人账户扣费
18
19mpirun hostname
20### 程序的执行命令

注意

上述中###为注释行。

第一行表示这是一个bash脚本,第4-17行以#SBATCH开头的命令表示这些是需要slurm系统处理的参数。

如下图所示,通过sbatch+作业脚本名提交作业,系统会返回作业编号,通过squeue命令可以看到作业运行状态,等作业执行完成后,默认会把程序的输出放到slurm-作业编号.out的文件中,可通过该文件查看程序的输出。

image6

GPU集群作业提交

如果是GPU集群,需要在作业脚本中增加--gres=gpu:<number of card>参数。例如#SBATCH --gres=gpu:2,意味着指定2张GPU卡数。

以下为GPU作业的一个示例:

 1#! /bin/bash
 2### 表示这是一个bash脚本
 3
 4#SBATCH --job-name=gpu-example
 5### 该作业的作业名
 6
 7#SBATCH --nodes=1
 8### 该作业需要1个节点
 9
10#SBATCH --ntasks=16
11### 该作业需要16个CPU
12
13#SBATCH --gres=gpu:1
14### 申请1块GPU卡
15
16#SBATCH --comment project_name
17### 指定从哪个项目扣费。如果没有这条参数,则从个人账户扣费
18
19source ~/.bashrc
20### 初始化环境变量
21
22python test.py
23### 程序的执行命令

注意

GPU集群中提交作业时,需要在srunsbatch命令中增加参数-s,或者 --oversubscribe。表示允许与其它作业共享资源。

例如:

1$sbatch -s job.sh

如果要在GPU集群中使用Nvidia指令,请参考GPU Cluster

常见提交作业参数参考

参数

说明

--job­name=<name>

设定作业名称

--nodes=<n>-N

设定作业需要的节点数。如果没有指定,默认分配足够的节点来满足--ntasks=<n>--cpus-per-task=<ncpus>参数的要求。

--ntasks-per-node=<ntasks>

设定每个节点上的任务数。要和 --nodes=<n> 同时配合使用。

--ntasks=<n>-n

设定最多启动的任务数。

--cpus-per-task=<ncpus>

设定每个任务所需要的CPU核数。如果没有指定,默认为每个任务分配一个CPU核。一般运行OpenMP等多线程程序时需要,普通MPI程序不需要。

--gres=gpu:n

设定需要使用的GPU卡数量

--comment projectName

设定需要扣费的项目账户,将 projectName 替换为项目名称。如果项目名称错误,作业会提交失败。

参考链接

Slurm作业调度系统使用指南 by 中国科大超级计算中心 李会民