Differences

This shows you the differences between two versions of the page.

--- techstaff:slurm [2018/12/07 12:35] – [Using the GPU] kauffman
+++ techstaff:slurm [2018/12/18 15:38] – kauffman
@@ Line 246: / Line 246: @@
 ====== Using the GPU ======
-[[ techstaff:cuda | Environment Variables and more ]]
+===== Environment Variables =====
-===== CUDA_VISIBLE_DEVICES =====
+==== CUDA_HOME, LD_LIBRARY_PATH ====
+Please make sure you specify $CUDA_HOME and if you want to take advantage of CUDNN libraries you will need to append /usr/local/cuda-x.x/lib64 to the $LD_LIBRARY_PATH environment variable.
+For example:
+  cuda_version=9.2
+  export CUDA_HOME=/usr/local/cuda-${cuda_version}
+  export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
+Currently we support the same versions of CUDA that the latest version of CUDNN supports. This is not written in stone and we can accommodate most other versions if required; just let techstaff know what your needs are.
+==== CUDA_VISIBLE_DEVICES ====
 Do not set this variable. It will be set for you by SLURM.