qsub_infer.sh

#!/bin/bash
# Usage: ./qsub_infer.sh [NUM_XPUS]
# Example: ./qsub_infer.sh 1

NUM_XPUS=${1:-1}

sbatch <<EOF
#!/bin/bash -l
#SBATCH --job-name=om-infer
#SBATCH --account=AIRR-P51-DAWN-GPU
#SBATCH --partition=pvc9
#SBATCH --nodes=1
#SBATCH --gres=gpu:${NUM_XPUS}
#SBATCH -n ${NUM_XPUS}
#SBATCH --time=10:00:00
#SBATCH --output=Logs/infer_%j.out
#SBATCH --error=Logs/infer_%j.err

. /etc/profile.d/modules.sh
module purge
module load rhel9/default-dawn

source ~/miniconda3/etc/profile.d/conda.sh
conda activate ~/rds/rds-airr-p51-TWhPgQVLKbA/Env/pub_env/pytorch-xpu

echo "Allocated ${NUM_XPUS} XPU(s)"
bash bash_infer.sh
EOF