计算集群使用说明¶
集群的基本概念¶
CPU/Core(核)的概念¶
CPU
是 Central Processing Unit 的缩写。比起全称,他的缩写更为大家所熟知。我们买电脑时都会看这个电脑拥有几个 CPU。CPU可以计算数字或者执行你的代码等等。每个CPU有多个计算核心(Core),调度系统可按照所需核心数对任务使用的资源进行分配,因而在实际使用中,我们常常用核
代替CPU
这个表述。
Memory(内存)的概念¶
内存
(Memory)就是储存数据的地方。跟硬盘(disk)储存的数据不同,内存
里的数据可以直接被 核
读取。跟你在硬盘里储存的数据类似,只是它被核
读取的速度更快。当执行程序时,有一些数据会先被读入内存
,然后再执行计算。因此内存越大,被读入的数据也就越多,能够同时处理的数据也就越多,代码运行的时间会更短。
Node(节点)的概念¶
节点
(Node)换个日常的说法就是你的电脑
,比如一台台式机或者笔记本电脑。它由若干个核
和一个内存
组成。因此可以把节点
简单理解成日常见到的电脑(主机)。
HPC(集群/超级计算机/超算)的概念¶
HPC
就是High Performance Cluster的缩写,又称为超级计算机,高性能集群等。它由若干个节点
组成。实际使用中,这些节点
会有不同的角色,通常包含登录节点
,管理节点
和计算节点
等。登录节点
顾名思义就是用来登录的节点
。用户从自己电脑可以登录到登录节点
。计算节点
是用来计算的节点,他们的唯一使命就是计算。管理节点
比较特殊,用来管理计算节点
,比如分配某某计算任务给某几个计算节点
来算。
Message Passing Interface(MPI)并行计算的概念¶
并行计算
是若干个节点
一起执行计算的意思。从节点
的概念可以知道,一个节点
的内存
和核
肯定是有限。比如,现有一个节点
有24个核
和32GB的内存
,我们想执行一个计算,用到48个核
,自然需要用到两个节点
。问题是另一个节点
的24个核
如何读取到第一个节点
的内存
里的数据?这一个时候就要用到MPI
/并行计算
了。MPI
是信息传输界面的简称。是一种告诉节点
怎么跨节点读取内存
的代码。也就是说这是计算机代码的一部分,我们常用的计算软件vasp
或cp2k
都已经写入了,所以只要直接使用便可以。
组内集群知识¶
本课题组使用 Zeus 计算集群提交计算任务进行计算模拟。Zeus 集群由两个登陆节点、一个管理节点、三个计算集群构成,每个计算集群包含多个计算节点(含六个 GPU 节点和一个大内存胖节点),其中 GPU 节点包括一个安装有 4 张 V100 的节点、一个安装有4张 A100 的节点和四个安装有 8 张 2080 Ti 的节点。
目前,所有 CPU 节点可以通过同一登陆节点进行提交,以下对集群使用的一些注意事项进行说明。关于 GPU 的使用,请参考使用集群上的GPU。
使用上述集群之前,你必须拥有一个账号才能进行任务提交。申请账号请联系集群管理员。
创建密钥对¶
Warning
新人必学
ssh
是用来安全进行登录远程电脑的命令。使用后,有两种选择来验证登录
- 使用密码
- 使用密钥
第一种方法已经为大众所熟知,但是不安全,目前集群对新开账号原则上不提供登陆密码。因此我们采用密钥进行登录。
使用如下命令生成密钥:
ssh-keygen
根据终端的提示进行操作(实际上你可能只需要不停按enter
键)。默认情况下你会在~/.ssh
目录中得到id_rsa
和id_rsa.pub
文件,他们分别是 私钥 和 公钥。创建好了之后请把 公钥 id_rsa.pub
文件发给服务器管理员。
Warning
私钥是登录集群的钥匙,请务必保管好这个文件,防止自己的电脑被入侵
获取账号¶
集群只允许已经授权的用户进行登录。在从管理员处获得你的账号名和初始密码后, Linux 或 Mac 用户可直接从命令行登录集群,使用 ssh
命令即可。
$ ssh -p <port> username@ip_address
请将 username
和 ip_address
替换为管理员提供的账号和IP地址,<port>
替换为端口号。
集群均采用 Linux 系统,因此不熟悉 Linux 基本操作的用户(例如查看文件、编辑文本、复制数据等)可以参考 Linux 入门,并熟悉这些操作。本文档假设用户有一定的 Linux 基础。
Windows 用户¶
对 Windows 用户来说,可以使用以下方法登陆集群。
-
(Windows 10/11用户推荐)使用 WSL(Windows Subsystem for Linux)。WSL 是 Windows 10 新版的特性,可使得用户在 Windows 系统下运行命令行模式的 Ubuntu 或 OpenSUSE 等子系统。使用 WSL 的用户可直接参考 Linux 的使用方法进行操作。具体安装方式可以参考官方教程。 对于使用集群的大多数需求,WSL 1 即可满足,因此不一定需要升级到 WSL 2 。
-
使用 Git Windows客户端,其自带一个基于Zsh的shell,亦可以提供对SSH的支持,体验更接近原生Bash,缺点是没有SFTP管理等功能。
-
使用 Xshell、PuTTY 等 SSH 客户端,Windows 10 以下的用户可使用这种方式。这类 SSH 客户端可以提供较完整的 SSH 功能。关于Putty的使用请参考。
-
使用虚拟机安装 Linux。若不想安装 Linux 双系统可以选择使用这种方式。正确配置的虚拟机和真正使用 Linux 几乎无差别。但虚拟机启动时间长,且完全启动时占用系统资源较多。
目录结构¶
Zeus 集群具有如下的目录结构,为了保持统一性,请在/data/username
(username
请替换为自己的用户名)下做计算。
/data <--目前的数据盘(432TB大存储)
├── 51-data <--原51备份后的数据
│ ├── ...
│ ├── ...
│ └── username
├── 52-data <--原52备份后的数据
│ ├── ...
│ ├── ...
│ └── username
├── home <--Zeus(191)登陆后的home文件夹
│ ├── ...
│ ├── ...
│ └── username
├── ...
├── ...
└── username <--在这里解压数据、提交计算
作业提交¶
计算节点、队列和脚本¶
通过sinfo
命令可以看到,目前的集群包括51/52/53三个类别,分别为51/52/53计算集群,51/52/53集群的计算节点分别对应编号为c51-00x/c52-00x/c53-00x
。
PARTITION AVAIL TIMELIMIT NODES STATE NODELIST
gpu1 up infinite 1 idle c51-g001
gpu2 up infinite 1 idle c51-g002
gpu3 up infinite 4 idle c51-m[001-004]
c51-small up 20:00 33 idle c51-[001-011,013-034]
c51-medium up 12:00:00 33 idle c51-[001-011,013-034]
c51-large up 1-00:00:00 33 idle c51-[001-011,013-034]
c51-long up 2-00:00:00 33 idle c51-[001-011,013-034]
c51-xlong up 3-00:00:00 33 idle c51-[001-011,013-034]
c51-xlarge up 1-00:00:00 33 idle c51-[001-011,013-034]
c51-exlong up 7-00:00:00 33 idle c51-[001-011,013-034]
c52-small up 20:00 40 idle c52-[001-040]
c52-medium up 12:00:00 40 idle c52-[001-040]
c52-large up 1-00:00:00 40 idle c52-[001-040]
c52-long up 2-00:00:00 40 idle c52-[001-040]
c52-xlong up 3-00:00:00 40 idle c52-[001-040]
c52-xlarge up 1-00:00:00 40 idle c52-[001-040]
c52-exlong up 7-00:00:00 40 idle c52-[001-040]
c53-small up 20:00 34 idle c53-[001-034]
c53-medium up 12:00:00 34 idle c53-[001-034]
c53-large up 1-00:00:00 34 idle c53-[001-034]
c53-long up 2-00:00:00 34 idle c53-[001-034]
c53-xlong up 3-00:00:00 34 idle c53-[001-034]
c53-xlarge* up 1-00:00:00 34 idle c53-[001-034]
由于处理器核数不同,任务只能在具有相同核数的节点间并行,由此对不同集群的节点按照队列进行了分组,队列前缀分别为51-
/52-
/53-
,其对应每个节点上的核数分别为24/28/32。通过sinfo
命令可以看到当前集群上的队列及其使用情况。
现编号为c51-00x
的节点需通过队列c51-small
、c51-medium
、c51-large
等等来进行提交,并设置核数为24的倍数(24,48,72等)以确定节点数,--ntasks-per-node=24
。 使用节点的数量通过总核数除以每个节点核数的值来确定。 同理,若想使用编号为c52-00x
的节点,则队列名为c52-small
、c52-medium
、c52-large
等等,核数为28的倍数(28,56,84等),--ntasks-per-node=28
;若想使用编号为c53-00x
的节点,则队列名为c53-small
、c53-medium
、c53-large
等等,核数为32的倍数(32,64,96等),--ntasks-per-node=32
。
GPU(Tesla V100节点)和胖节点仍按照51进行编组,编号分别为
c51-g001
和c51-s001
。
目前每个队列仍限制同时运行4个任务、队列内使用至多12个节点。新增全局任务限制,即三组队列总共使用核数不超过556,若超出此限制则任务会处于PEND状态。
提交脚本示例放在/data/share/base/scripts
里面,软件统一安装在/data/share/apps
下,目前安装了VASP 5.4.4、CP2K 7.1、Gaussian 16、Lammps、Gromacs、DeePMD-kit等。
这里对作业提交脚本举例说明如下:
#!/bin/bash
#SBATCH -J cp2k
#SBATCH -o cp2k.out.%j
#SBATCH -e cp2k.err.%j
#SBATCH -p c53-large
#SBATCH -N 2
#SBATCH --ntasks-per-node=32
#SBATCH --exclusive
#SBATCH --mem=8G
# add modulefiles
ulimit -s unlimited
module load intel/17.5.239 mpi/intel/2017.5.239
module load gcc/5.5.0
module load cp2k/7.1
mpiexec.hydra cp2k.popt input.inp >& output_$LSB_JOBID
其中:
#SBATCH -p 队列名
用于指定作业提交的队列。#SBATCH -t hh:mm:ss
用于指定任务所需的时间(Walltime),若运行超过hh:mm:ss
,则任务会被管理系统杀死。对于不同类型的队列,Walltime上限有所不同。对small
队列要求在20分钟以内,对medium
要求在12小时以内,对large
和xlarge
要求在24小时以内,对long
要求在48小时以内,对xlong
则在72小时以内。#SBATCH --job-name=cp2k
指定作业名称,一般按照实际计算来取名以方便查看完成情况。#SBATCH -N 2
指定作业提交的总节点数,#SBATCH --ntasks-per-node=32
指定提交队列的每个节点上的CPU总核数,例如这里在53队列中选取2个节点进行并行计算,即使用了64个核。#SBATCH --mem=8G
指定作业所需消耗的内存,例如这里限制作业占用内存为 8 GB。module load xxx
用于加载环境,保持/data/share/base/scripts
示例中的写法即可。mpiexec.hydra cp2k.popt input.inp >& output_$LSB_JOBID
是实际执行任务的命令。
可以看到,任务提交脚本实际上是一个具有特殊注释格式的 bash 脚本。因此在加载环境后,可以使用 bash 语法来设置环境变量、控制任务运行的路径、进行批处理等等。
注意
对于 fat
和 gpu*
队列,请 务必 指定作业所需的内存不小于自己任务实际需要内存大小!
否则可能会因为其他人任务尚未结束而迟迟无法运行,或因为申请了过多的内存资源而使得其他任务无法提交。
为了公平使用,请一定遵守上述规则。 管理员会根据用户反馈直接清除未遵循规范的任务。
相应地,其他计算队列通常会一个任务独占若干节点,因而不需要设置内存,保持默认——占满即可。 考虑到 DFT 计算本身对内存有一定需求,请务必注意自己的设置符合实际情况,以免浪费宝贵的时间和机时。
作业提交¶
若用户已经准备好相应计算的输入和提交脚本,则可以对任务进行提交。例如提交脚本文件名为cp2k.slurm
,则提交命令为:
sbatch cp2k.slurm
若提交成功,可以看到以下提示:
Job <1360> is submitted to queue <53-large>
表示任务已经成功提交到节点上,编号为 1360
。
任务提交后,可以通过squeue -u <username>
命令查看自己任务的运行情况,<username>
即自己的用户名。
JOBID USER STAT QUEUE FROM_HOST EXEC_HOST JOB_NAME SUBMIT_TIME
1227 user RUN 52-medium mgt02 28*c52-032 CoX Mar 9 22:35
28*c52-023
1133 user RUN 51-medium mgt02 24*c51-024 Cu13 Mar 9 21:20
24*c51-031
1360 user PEND 53-large mgt02 cp2k Mar 10 13:26
其中 JOBID
即为任务编号,STAT
表示状态,RUN
即为正在运行,而 PEND
表示正在排队,可能是因为空余节点数不足。可以看到,1227和1133号任务正在运行,分别使用了2个节点,刚刚提交的1360号任务则在排队。
如果想要停止或取消已经提交的任务,则使用命令:
bkill 1360
若看到 Job <1360> is being terminated
的提示,则说明停止任务的请求已经发出。一段时间后,该任务即被杀死。
链接
- 更多使用教程和说明请参考:Slurm作业调度系统使用指南
- 关于Slurm系统和其他调度系统间的命令对照,请参考官方提供的 Rosetta石碑
登出集群¶
请在命令行中输入:
exit
回车即可退出登陆。
校外访问¶
若为在校师生,可使用学校提供的 SSLVPN 登陆集群。
详细配置方法请参阅:SSLVPN 使用说明-厦门大学VPN专题网站。