summaryrefslogtreecommitdiffstats
path: root/roles/cuda/templates/cuda_init.service.j2
diff options
context:
space:
mode:
Diffstat (limited to 'roles/cuda/templates/cuda_init.service.j2')
-rw-r--r--roles/cuda/templates/cuda_init.service.j213
1 files changed, 13 insertions, 0 deletions
diff --git a/roles/cuda/templates/cuda_init.service.j2 b/roles/cuda/templates/cuda_init.service.j2
new file mode 100644
index 0000000..c61cbad
--- /dev/null
+++ b/roles/cuda/templates/cuda_init.service.j2
@@ -0,0 +1,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