diff options
| author | Suren A. Chilingaryan <csa@suren.me> | 2018-07-05 16:22:58 +0200 |
|---|---|---|
| committer | Suren A. Chilingaryan <csa@suren.me> | 2018-07-05 16:22:58 +0200 |
| commit | 0b3d99f96dabb1aa2fbf7f99559852916a97d610 (patch) | |
| tree | b56e453cf5182a24c6878c23f5985781928d1e57 /roles/cuda/templates/cuda_init.sh.j2 | |
| parent | 5416f8599ffa92b69bb87434dc1175d760760bbe (diff) | |
| parent | 813756495d5ef33fa3cc95d69b6d88418ebe7bb1 (diff) | |
| download | itm-0b3d99f96dabb1aa2fbf7f99559852916a97d610.tar.gz itm-0b3d99f96dabb1aa2fbf7f99559852916a97d610.tar.bz2 itm-0b3d99f96dabb1aa2fbf7f99559852916a97d610.tar.xz itm-0b3d99f96dabb1aa2fbf7f99559852916a97d610.zip | |
Merge commit '813756495d5ef33fa3cc95d69b6d88418ebe7bb1' as 'roles/cuda'
Diffstat (limited to 'roles/cuda/templates/cuda_init.sh.j2')
| -rw-r--r-- | roles/cuda/templates/cuda_init.sh.j2 | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/roles/cuda/templates/cuda_init.sh.j2 b/roles/cuda/templates/cuda_init.sh.j2 new file mode 100644 index 0000000..3c85ea2 --- /dev/null +++ b/roles/cuda/templates/cuda_init.sh.j2 @@ -0,0 +1,9 @@ +#!/bin/bash +# During boot we initialize the GPUs (creates /dev/nvidia*) + +if [ ! -f "/usr/bin/nvidia-smi" ]; then + logger -s -t nvidia-smi "Script $0 could not find /usr/bin/nvidia-smi" +else + /usr/bin/nvidia-smi --compute-mode={{ cuda_init_compute_mode }} + /usr/bin/nvidia-smi --persistence-mode={{ cuda_init_persistence_mode }} +fi |