File tree Expand file tree Collapse file tree 1 file changed +9
-11
lines changed
Expand file tree Collapse file tree 1 file changed +9
-11
lines changed Original file line number Diff line number Diff line change 11#! /bin/bash
2- # SBATCH -t 01:30 :00
2+ # SBATCH -t 01:00 :00
33# SBATCH -N 3
44# SBATCH --ntasks-per-node=4
55# SBATCH --ntasks-per-socket=2
66# SBATCH --gres=gpu:4
77# SBATCH -c 4
88# SBATCH --mem-per-cpu=0
99
10- export PYTHONHASHSEED=0
11- module load anaconda
12- source activate pppl
10+ module load anaconda/4.4.0
11+ source activate PPPL
1312module load cudatoolkit/8.0
1413module load cudnn/cuda-8.0/6.0
1514module load openmpi/cuda-8.0/intel-17.0/2.1.0/64
16- module load intel/17.0/64/17.0.4.196 intel-mkl/2017.3/4/64
15+ module load intel/17.0/64/17.0.4.196
1716
1817# remove checkpoints for a benchmark run
19- rm /tigress /$USER /model_checkpoints/*
20- rm /tigress /$USER /results/*
21- rm /tigress /$USER /csv_logs/*
22- rm /tigress /$USER /Graph/*
23- rm /tigress /$USER /normalization/*
18+ rm /scratch/gpfs /$USER /model_checkpoints/*
19+ rm /scratch/gpfs /$USER /results/*
20+ rm /scratch/gpfs /$USER /csv_logs/*
21+ rm /scratch/gpfs /$USER /Graph/*
22+ rm /scratch/gpfs /$USER /normalization/*
2423
2524export OMPI_MCA_btl=" tcp,self,sm"
26-
2725srun python mpi_learn.py
You can’t perform that action at this time.
0 commit comments