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 @@
# 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
......
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