script.sh
No OneTemporary
Actions

Subscribers

None

File Metadata

	#!/bin/bash

	#SBATCH --nodes=1
	#SBATCH --time=1:0:0
	#SBATCH --partition=gpu
	#SBATCH --gres=gpu:1
	#SBATCH --qos=gpu

	source /ssoft/spack/bin/slmodules.sh -s x86_E5v2_Mellanox_GPU
	source /scratch/nvarini/tensorflow-1.9.0-pip3/bin/activate

	module load gcc/6.4.0 cuda/9.1.85 openmpi/3.0.1-cuda python cudnn
	export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
	srun python tf_cnn_benchmarks.py --batch_size=64 --model=resnet50 --num_gpus=1 --data_format=NCHW --variable_update=replicated --local_parameter_device=gpu --num_batches=3000

	deactivate