#!/bin/bash

export PYTHONPATH=`pwd`/minigo/bazel-bin/cc:$PYTHONPATH
source config_DGXA100.sh

SLURM_NTASKS_PER_NODE=$(( 1 + PROCS_PER_GPU * DGXNGPU ))
export SLURM_NTASKS_PER_NODE

mpirun --allow-run-as-root -np $SLURM_NTASKS_PER_NODE ./run.sh
