6-global_knn.slurm

#!/bin/bash

#SBATCH --wait-all-nodes=1
#SBATCH --nodes=1
#SBATCH --cpus-per-task=16
#SBATCH --ntasks-per-node=1
#SBATCH --exclusive
#SBATCH --output /apps/aws-distributed-training-workshop-pcluster/head-node-scripts/knn_out_%j.out
#SBATCH --error /apps/aws-distributed-training-workshop-pcluster/head-node-scripts/knn_err_%j.err

source /home/ec2-user/.bashrc
conda activate pytorch-py38

cd /apps/aws-distributed-training-workshop-pcluster/head-node-scripts/pyscripts/

srun python3 global_knn.py