roles/cuda/templates/cuda_init.service.j2


1
2
3
4
5
6
7
8
9
10
11
12
13

[Unit]
Description=Initialize nvidia/cuda devices
Before=slurm.service
DefaultDependencies=no
Conflicts=shutdown.target

[Service]
ExecStart=/usr/local/bin/cuda_init.sh
Type=oneshot
RemainAfterExit=yes

[Install]
WantedBy=multi-user.target