slurm调用GPU运算的脚本案例

以下是某程序需要GPU运算，slurm的脚本示例

#!/bin/bash
#SBATCH -J test  #作业名称
#SBATCH -P gpuq  #分区
#SBATCH -o %j.out  #输出结果文件
#SBATCH -e %j.err  #输出错误文件
#SBATCH -N 1  #节点数量
#SBATCH --ntasks-per-node=1   #每个节点的进程数
#SBATCH --gres=gpu:1  #一个GPU资源

module load WRF   #加载计算程序
#具体的程序计算参数