launch_model.sh
51f5c1e03170MPI_hyper_plus_horovod

launch_model.sh
View Options

			#!/bin/bash

			## ====== Settings


			## specific for GPU nodes (free account)
			#SBATCH --partition=gpu --qos=gpu --gres=gpu:2

			#SBATCH --nodes=1
			#SBATCH --ntasks-per-node=2
			#SBATCH --cpus-per-task=1
			#SBATCH --mem=60G
			#SBATCH --time=1:00:00

			echo STARTING AT $(date)

			## == Specify here the needed modules and the virtual env

			module purge
			source /ssoft/spack/bin/slmodules.sh -s x86_E5v2_Mellanox_GPU
			module load gcc cuda cudnn python mvapich2
			module list
			source /home/nvarini/tensorflow-1.13.1-venv/bin/activate

			## == Specify here the commands to execute
			export CUDA_VISIBLE_DEVICES=0
			## Run. Argument to pass (in order): rotations, lights, dice, weightnopv, weightpv, epochs
			export MV2_ENABLE_AFFINITY=0
			srun --export=ALL --cpu_bind=verbose,rank python model_main.py
			#srun python mpi4py_pycuda_demo.py

			deactivate

			echo FINISHED AT $(date)