Skip to content
Snippets Groups Projects
Commit 62ffb7a5 authored by patmjen's avatar patmjen
Browse files

Change config variable names in init script

parent c12bbdfc
No related branches found
No related tags found
No related merge requests found
...@@ -4,31 +4,31 @@ ...@@ -4,31 +4,31 @@
# Configuration # Configuration
# This is what you should change for your setup # This is what you should change for your setup
PYVER=3.9.14 # Python version PYTHON_VERSION=3.9.14 # Python version
CUVER=11.6 # CUDA version CUDA_VERSION=11.6 # CUDA version
VENVDIR=. # Where to store your virtualenv VENV_DIR=. # Where to store your virtualenv
VENVNAME=venv # Name of your virtualenv VENV_NAME=venv # Name of your virtualenv
# Load modules # Load modules
module load python3/$PYVER module load python3/$PYTHON_VERSION
module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "numpy/") module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "numpy/")
module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "scipy/") module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "scipy/")
module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "matplotlib/") module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "matplotlib/")
module load $(module avail -o modulepath -t -C "python-${PYVER}" | grep "pandas/") module load $(module avail -o modulepath -t -C "python-${PYTHON_VERSION}" | grep "pandas/")
module load cuda/$CUVER module load cuda/$CUDA_VERSION
CUDNN_MOD=$(module avail -o modulepath -t cudnn | grep "cuda-${CUVER}" | sort | tail -n1) CUDNN_MOD=$(module avail -o modulepath -t cudnn | grep "cuda-${CUDA_VERSION}" | sort | tail -n1)
if [[ ${CUDNN_MOD} ]] if [[ ${CUDNN_MOD} ]]
then then
module load ${CUDNN_MOD} module load ${CUDNN_MOD}
fi fi
# Create virtualenv if needed and activate it # Create virtualenv if needed and activate it
if [ ! -d "${VENVDIR}/${VENVNAME}" ] if [ ! -d "${VENV_DIR}/${VENV_NAME}" ]
then then
echo INFO: Did not find virtualenv. Creating... echo INFO: Did not find virtualenv. Creating...
virtualenv ${VENVDIR}/${VENVNAME} virtualenv ${VENV_DIR}/${VENV_NAME}
fi fi
source ${VENVDIR}/${VENVNAME}/bin/activate source ${VENV_DIR}/${VENV_NAME}/bin/activate
# Make all GPUs visible # Make all GPUs visible
if command -v nvidia-smi &> /dev/null if command -v nvidia-smi &> /dev/null
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment