test-can-1-2 / run-inference.sh
rdeinla's picture
End of training
3fa2f59 verified
#!/bin/sh
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=4
#SBATCH --cpus-per-task=4
#SBATCH --mem=15GB
#SBATCH --time=0-10:00
#SBATCH --partition=livi
#SBATCH --mail-user=deinlar@myumanitoba.ca
#SBATCH --mail-type=END
#SBATCH --gpus=1
module load cuda/12.4.1 arch/avx2 gcc/13.2.0
module load python/3.11.11
source /project/6085367/SharedData/my_newvenv/bin/activate
python3 inference.py>log.txt