LebwohlLasher/ll_vector_array_np.sh at main · RedOffset6/LebwohlLasher · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#!/bin/bash

#SBATCH --account=PHYS030544
#SBATCH --job-name=ll_mpi_numba_array
#SBATCH --partition=teach_cpu
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --mem-per-cpu=5GB
#SBATCH --time=0:30:00
#SBATCH --array=1-10

echo 'running mpi test'

module load languages/miniconda

echo "Before activation: $(which python)"

#/mnt/storage/software/languages/miniconda/miniconda3.11.3-CompressAI-Vision0.1.3/condabin/conda activate desktop_clone_env
#conda init bash
#possibly I need to run "conda init bash" every time i open a new terminal
source activate desktop_clone_env

echo "After activation: $(which python)"

#module load languages/intel/2020-u4

size=$((SLURM_ARRAY_TASK_ID*75))

srun --mpi=pmi2 python LebwohlLasher_vector_np.py 50 $size 0.5 0