#!/bin/sh #SBATCH --job-name=HPQ_job_name # Job name #SBATCH --ntasks=1 # Run on a single CPU #SBATCH --time=48:00:00 # Time limit hrs:min:sec #SBATCH --output=serial_test_%j.out # Standard output and error log #SBATCH --gres=gpu:4 #SBATCH --partition=hpq_2day_4G pwd; hostname; date | tee result nvidia-docker run -it ${USE_TTY} --name $SLURM_JOB_ID --user $(id -u $USER):$(id -g $USER) --rm -v </home or /localscratch path>:/workspace -v /etc/passwd:/etc/passwd -v /etc/group:/etc/group -v /etc/shadow:/etc/shadow nvcr.io/nvidia/pytorch:18.07-py3 python -c 'import torch; print(torch.__version__)'