#! /bin/bash

host=`hostname`

if [[ "$NERSC_HOST" == cori ]]; then
    echo \$NERSC_HOST matched cori
    module load cray-python/3.8.5.0

    which python
elif [[ "$NERSC_HOST" == perlmutter ]]; then
    echo \$NERSC_HOST matched perlmutter

    module load cray-python/3.9.4.1

    which python
fi