From 62ffb7a53ec9e6a0fe623f145790c5c9d8779973 Mon Sep 17 00:00:00 2001 From: "Patrick M. Jensen" <patmjen@dtu.dk> Date: Wed, 16 Nov 2022 11:03:18 +0100 Subject: [PATCH] Change config variable names in init script --- scripts/init.sh | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/scripts/init.sh b/scripts/init.sh index 3dd2b08..043db73 100644 --- a/scripts/init.sh +++ b/scripts/init.sh @@ -4,31 +4,31 @@ # Configuration # This is what you should change for your setup -PYVER=3.9.14 # Python version -CUVER=11.6 # CUDA version -VENVDIR=. # Where to store your virtualenv -VENVNAME=venv # Name of your virtualenv +PYTHON_VERSION=3.9.14 # Python version +CUDA_VERSION=11.6 # CUDA version +VENV_DIR=. # Where to store your virtualenv +VENV_NAME=venv # Name of your virtualenv # Load modules -module load python3/$PYVER -module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "numpy/") -module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "scipy/") -module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "matplotlib/") -module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "pandas/") -module load cuda/$CUVER -CUDNN_MOD=$(module avail -o modulepath -t cudnn | grep "cuda-${CUVER}" | sort | tail -n1) +module load python3/$PYTHON_VERSION +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "numpy/") +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "scipy/") +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "matplotlib/") +module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "pandas/") +module load cuda/$CUDA_VERSION +CUDNN_MOD=$(module avail -o modulepath -t cudnn | grep "cuda-${CUDA_VERSION}" | sort | tail -n1) if [[ ${CUDNN_MOD} ]] then module load ${CUDNN_MOD} fi # Create virtualenv if needed and activate it -if [ ! -d "${VENVDIR}/${VENVNAME}" ] +if [ ! -d "${VENV_DIR}/${VENV_NAME}" ] then echo INFO: Did not find virtualenv. Creating... - virtualenv ${VENVDIR}/${VENVNAME} + virtualenv ${VENV_DIR}/${VENV_NAME} fi -source ${VENVDIR}/${VENVNAME}/bin/activate +source ${VENV_DIR}/${VENV_NAME}/bin/activate # Make all GPUs visible if command -v nvidia-smi &> /dev/null -- GitLab