22 lines
533 B
Bash
22 lines
533 B
Bash
#!/bin/sh
|
|
#To be used before srun so that interactive sessions are run with gpu support
|
|
export CUDA_HOME=/opt/cuda-8.0.44
|
|
|
|
export CUDNN_HOME=/opt/cuDNN-6.0_8.0
|
|
|
|
export STUDENT_ID=$(whoami)
|
|
|
|
export LD_LIBRARY_PATH=${CUDNN_HOME}/lib64:${CUDA_HOME}/lib64:$LD_LIBRARY_PATH
|
|
|
|
export LIBRARY_PATH=${CUDNN_HOME}/lib64:$LIBRARY_PATH
|
|
|
|
export CPATH=${CUDNN_HOME}/include:$CPATH
|
|
|
|
export PATH=${CUDA_HOME}/bin:${PATH}
|
|
|
|
export PYTHON_PATH=$PATH
|
|
|
|
# Activate the relevant virtual environment:
|
|
|
|
source /home/${STUDENT_ID}/miniconda3/bin/activate mlp
|