launch_model.sh
No OneTemporary
Actions

Subscribers

None

File Metadata

	#!/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)