nixos/devices/xmupc1
2024-06-14 12:07:17 +08:00
..
secrets
default.nix
README.md

硬件

  • CPU16 核 32 线程。
  • 内存96 G。
  • 显卡:
    • 409024 G 显存。
    • 309024 G 显存。
    • 2080Ti: 12 G 显存。
  • 硬盘2 T。

队列系统SLURM

基本概念

SLURM 是一个用来对任务排队的系统,轮到某个任务时,再调用其它程序来执行这个任务。

常用命令

我做了一个 TUI 界面,用起来比较简单,大多情况下可以满足需求。命令为:

sbatch-tui

sbatch

如果需要在提交任务时指定更详细的细节,或者要编写脚本批量提交任务,则在 sbatch 后面加上参数,这时是直接调用来自 SLURM 的 sbatch 命令。 常用的参数见下文。更详细的内容见 SLURM 的官方文档。

提交一个 VASP GPU 任务的例子:

sbatch --gpus=1 --ntasks-per-gpu=1 --job-name="my great job" vasp-nvidia
  • --gpus 指定使用GPU 的情况:
    • 要占用任意一个 GPU排到这个任务时哪个空闲就使用哪个--gpus=1。要占用任意两个就写 --gpus=2,以此类推。 但一般来说,单个任务不要占用超过一个 GPU,多个显卡的速度会比单个更慢。
    • 要指定具体使用哪个 GPU 时,写 --gpus=4090:1。2080 Ti 需要写为 2080_tiP5000 需要写为 p5000
    • 当需要使用多个不同类型的显卡(例如,指定使用一个 3090 和一个 4090--gres=gpu:3090:1,gpu:4090:1
  • --ntasks-per-gpu=1 对于 VASP 来说一定要写。
  • --job-name=xxx 指定任务的名字。可以简写为 -J。也可以不指定。
  • 默认情况下,一个 task 会搭配分配一个 CPU 核(一个线程),一般已经够用。如果一定要修改,用 --cpus-per-task
  • vasp-nvidia 指调用 std 版本,要使用 gam 或 ncl 版本时,写为例如 vasp-nvidia-gam

提交一个 VASP CPU 任务的例子:

sbatch --ntasks=4 --cpus-per-task=4 --hint=nomultithread --job-name="my great job" vasp-intel
  • --ntasks=4 --cpus-per-task=4 指定使用占用多少核。
    • CPU 的调度是个非常复杂的问题,而且 slurm 和 Intel MPI 之间的兼容性也不算好,因此推荐照抄下面的设置。 也可以自己测试一下怎样分配更好,但不要随意地设置。不同的设置会成倍地影响性能。
      • 对于 xmupc1--ntasks=3 --cpus-per-task=4
      • 对于 xmupc2--ntasks=4 --cpus-per-task=10
  • --hint=nomultithread 记得写。
  • --job-name=xxx 指定任务的名字。可以简写为 -J。也可以不指定。
  • vasp-intel 指调用 std 版本,要使用 gam 或 ncl 版本时,写为例如 vasp-intel-gam

要把其它程序提交到队列里,也是类似的写法。请自行举一反三。

要列出已经提交(包括已经完成、取消、失败)的任务:

squeue -t all -l

取消一个任务:

# 按任务的 id 取消
scancel 114514
# 按任务的名字取消
scancel -n my_great_job
# 取消一个用户的所有任务
scancel -u chn

要将自己已经提交的一个任务优先级提到最高(相应降低其它任务的优先级,使得总体来说不影响别人的任务):

scontrol top 114514

要显示一个任务的详细信息(不包括服务器重启之前算过的任务):

scontrol show job 114514

要显示一个任务的详细信息(包括服务器重启之前算过的任务):

sacct --units M --format=ALL -j 114514 | bat -S

sbatch 的更多参数

# 提交一个新任务,但是礼让后面的任务(推迟到指定时间再开始排队)
--begin=16:00 --begin=now+1hour
# 指定工作目录
--chdir=/path/to/your/workdir
# 指定备注
--comment="my great job"
# 指定任务的 ddl算不完就杀掉
--deadline=now+1hour
# 标准错误输出写到别的文件里
--error=error.log
# 将一些环境变量传递给任务(=ALL是默认行为
--export=ALL,MY_ENV_VAR=my_value
# 不传递现在的环境变量
--export=NONE
# 打开一个文件作为标准输入
--input=
# 发生一些事件(任务完成等)时发邮件
--mail-type=NONE,BEGIN,END,FAIL,REQUEUE,ALL --mail-user=chn@chn.moe
# 要求分配内存(不会真的限制内存使用,只是在分配资源时会考虑)
--mem=20G --mem-per-cpu --mem-per-gpu
# 输出文件是否覆盖
--open-mode={append|truncate}
# 指定输出文件
-o, --output=<filename_pattern>
# 不排队,直接跑(超额分配)
-s, --oversubscribe
# 包裹一个二进制程序
--wrap=

支持的连接协议

SSH

ssh 就是 putty winscp 之类的工具使用的那个协议。

  • 地址xmupc1.chn.moe
  • 端口6007
  • 用户名:自己名字的拼音首字母
  • 可以用密码登陆,也可以用证书登陆。

从一台服务器登陆到其它服务器,只需要使用 `ssh`` 命令:

ssh jykang
ssh xmupc1
ssh xmupc2
ssh user@host

直接从另外一台服务器下载文件,可以使用 rsync 命令:

rsync -avzP jykang:/path/to/remote/directory_or_file /path/to/local/directory

将另外一个服务器的某个目录挂载到这个服务器,可以使用 sshfs 命令:

sshfs jykang:/path/to/remote/directory /path/to/local/directory

用完之后记得卸载(不卸载也不会有什么后果,只是怕之后忘记了以为这是本地的目录,以及如果网络不稳定的话,运行在这里的软件可能会卡住):

umount /path/to/local/directory

如果不喜欢敲命令来挂载/卸载远程目录,也可以 RDP 登陆后用 dolphin。

RDP

就是 windows 那个远程桌面。

  • 地址xmupc1.chn.moe
  • 用户名:自己名字的拼音首字母
  • 密码和 ssh 一样(使用同样的验证机制)。

RDP 暂时没有硬件加速(主要是毛玻璃之类的特效会有点卡)。

记得在连接时点击“显示选项”将“体验”中的连接速度改为“LAN10 Mbps 或更高)”,不然会很卡。

samba

samba 就是 windows 共享文件夹的那个协议。

  • 地址xmupc1.chn.moe
  • 用户名:自己名字的拼音首字母
  • 初始密码和 ssh 一样,你可以自己修改密码(使用 smbpasswd 命令。samba 的密码和 ssh/rdp 的密码是分开的,它们使用不同的验证机制。

在 windows 上,可以直接在资源管理器中输入 \\xmupc1.chn.moe 访问。 也可以将它作为一个网络驱动器添加(地址同样是 \\xmupc1.chn.moe)。

计算软件

VASP

VASP 有很多很多个版本,具体来说:

  • VASP 可以用不同的编译器编译。目前安装的有nvidia、intel。nvidia 使用 GPU 计算intel 能用 CPU 计算。其它版本性能不佳,没有安装。
  • VASP 的 std/gam/ncl 版本有一点区别,一般用 std只有一个 gamma 点的时候用 gam 会快一点,系统中存在方向不平行的磁矩时必须用 ncl。
  • 无论哪个版本,都集成了下面这些补丁:
    • HDF5用于生成 hdf5 格式的输出文件。
    • wannier90我也不知道干啥的随手加上的。
    • OPTCELL如果存在一个 OPTCELL 文件VASP 会据此决定弛豫时仅优化哪几个晶胞参数。
    • MPI shared memory用来减小内存占用。
    • VTST tools用来计算 neb。

如何提交 VASP 到队列系统已经在上面介绍过了。下面的例子是,如果要直接运行一个任务的写法:

vasp-nvidia-env mpirun -np 1 -x CUDA_DEVICE_ORDER=PCI_BUS_ID -x CUDA_VISIBLE_DEVICES=0 -x OMP_NUM_THREADS=4 vasp-std
vasp-intel-env mpirun -n 2 -genv OMP_NUM_THREADS=4 vasp-std

其中 CUDA_VISIBLE_DEVICES 用于指定用哪几个显卡计算(多个显卡用逗号分隔)。 要查看显卡的编号,可以用 CUDA_DEVICE_ORDER=PCI_BUS_ID vasp-nvidia-env nvaccelinfo 命令。

这里 vasp-xxx-env 命令的作用是,进入一个安装了对应版本的 VASP 的环境,实际上和 VASP 关系不大; 后面的 mpirun xxx 才是真的调用 VASP。 所以实际上你也可以在这个环境里做别的事情,例如执行上面的 nvaccelinfo 命令。

要使用 VTST tools 里带的脚本,需要在命令前加上 vtstscripts 。例如:

vtstscripts dist.pl POSCAR.init POSCAR.final

mumax

问龚斌,我没用过。

lammps

除了我应该没人用,就不写了。

quantum espresso

我也只用过一次。大规模用到了再说吧。

其它软件

我自己电脑上有的软件,服务器都有装,用于科研的比如 VESTA 什么的。可以自己去菜单里翻一翻。

操作系统

操作系统是 NixOS是一个相对来说比较小众的系统。 它是一个所谓“函数式”的系统。 也就说,理想情况下,系统的状态(包括装了什么软件、每个软件和服务的设置等等)是由一组配置文件唯一决定的(这组配置文件放在 /etc/nixos 中)。 要修改系统的状态(新增软件、修改设置等等),只需要修改这组配置文件,然后要求系统应用这组配置文件就可以了, 系统会自动计算出应该怎么做(增加、删除、修改哪些文件,重启哪些服务等等)。 这样设计有许多好处,例如可以方便地回滚到之前任意一个时刻的状态(方便在调试时试错); 一份配置文件可以描述多台机器的系统,在一台上调试好后在其它机器上直接部署; 以及适合抄或者引用别人写好的配置文件。

以上都是对于管理员来说的好处。对于用户来说的好处不是太多,但是也有一些。 举个例子,如果用户需要使用一个没有安装的软件(例如 phonopy,当然实际上这个已经装了),只需要在要执行的命令前加一个逗号:

, phonopy --dim 2 2 2

系统就会帮你下载所有的依赖,并在一个隔离的环境中运行这个命令(不会影响这之后系统的状态)。

还有一个命令可能也有用,叫 try。 它会在当前的文件系统上添加一个 overlay之后执行的命令对文件的修改只会发生在这个 overlay 上; 命令执行完成后,它会告诉你哪些文件发生了改变,然后可以选择实际应用这些改变还是丢弃这些改变。 例如:

try phonopy --dim 2 2 2

这个命令和 NixOS 无关,只是突然想起来了。

文件系统

文件系统是 BtrFS。它的好处有

  • 同样的内容只占用一份空间;以及内容会被压缩存储(在读取时自动解压)。这样大致可以节省一半左右的空间。 例如现在 xll 目录里放了 213 G 文件,但只占用了 137 G 空间。
  • 每小时自动备份,放置在 /nix/persistent/.snapshots 中,大致上会保留最近一周的备份。如果你误删了什么文件,可以去里面找回。

ZSH

所谓 “shell” 就是将敲击的一行行命令转换成操作系统能理解的系统调用C 语言的函数)的那个东西,也就是负责解释敲进去的命令的意思的那个程序。

大多情况下默认的 shell 是 bash但我装的服务器上用 zsh。 zsh 几乎完全兼容 bash 的语法,除此以外有一些顺手的功能:

  • 如果忘记了曾经输入过的一个命令,输入其中的几个连续的字母或者单词(不一定是开头的几个字母),然后按 键,就会自动在历史命令中依次搜索。 例如我输入 install 按几下 键,就可以找到 sudo nixos-rebuild boot --flake . --install-bootloader --option substituters https://nix-store.chn.moe 这个东西。
  • 如果从头开始输入一个曾经输入过的命令,会用浅灰色提示这个命令。要直接补全全部命令,按 键。要补全一个单词,按 Ctrl + 键。
  • 常用的命令,以及常用命令的常用选项,按几下 tab 键,会自动补全或者弹出提示。