请问,我可不可在这里调用DCU计算资源,使用deepmd-kit版本的lammps进行运算处理呢?如果方便的话,可以给一下slurm的提交任务模板吗?谢谢
您好,可以的。
以下提交任务模板,以安装的deepmd-kit的c++接口为例。
(1)首先,在提交算例的in文件中,使用
plugin load libdeepmd_lmp.so ### 请更新so文件的路径
(2)执行lammps执行语句
#!/bin/bash
#SBATCH -J LAMMPS
#SBATCH -p wzhdnormal ### update 队列号
#SBATCH -N 1
#SBATCH -n 32
#SBATCH --gres=dcu:4 ### 每个节点 4 卡
#SBATCH -o LAMMPS-TEST.%j.out
### exampl:
您好,可以的。
以下提交任务模板,以安装的deepmd-kit的c++接口为例。
(1)首先,在提交算例的in文件中,使用
plugin load libdeepmd_lmp.so ### 请更新so文件的路径
(2)执行lammps执行语句
#!/bin/bash
#SBATCH -J LAMMPS
#SBATCH -p wzhdnormal ### update 队列号
#SBATCH -N 1
#SBATCH -n 32
#SBATCH --gres=dcu:4 ### 每个节点 4 卡
#SBATCH -o LAMMPS-TEST.%j.out
### exampl:
mpirun -np 4 lmp_mpi -k on g 4 -sf kk -pk kokkos gpu/aware off -in in.lj