techstaff:slurm
Differences
This shows you the differences between two versions of the page.
Both sides previous revisionPrevious revisionNext revision | Previous revisionNext revisionBoth sides next revision | ||
techstaff:slurm [2018/11/21 11:31] – [$CUDA_VISIBLE_DEVICES] kauffman | techstaff:slurm [2018/12/18 15:38] – kauffman | ||
---|---|---|---|
Line 246: | Line 246: | ||
====== Using the GPU ====== | ====== Using the GPU ====== | ||
+ | ===== 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 / | ||
+ | |||
+ | For example: | ||
+ | |||
+ | cuda_version=9.2 | ||
+ | export CUDA_HOME=/ | ||
+ | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH: | ||
+ | |||
+ | 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. | Do not set this variable. It will be set for you by SLURM. | ||
/var/lib/dokuwiki/data/pages/techstaff/slurm.txt · Last modified: 2021/01/06 16:13 by kauffman