templates/cuda_init.sh.j2


1
2
3
4
5
6
7
8
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