From 813756495d5ef33fa3cc95d69b6d88418ebe7bb1 Mon Sep 17 00:00:00 2001
From: "Suren A. Chilingaryan" <csa@suren.me>
Date: Thu, 5 Jul 2018 16:22:58 +0200
Subject: Squashed 'roles/cuda/' content from commit f82a4fe
git-subtree-dir: roles/cuda
git-subtree-split: f82a4fedb62a410b1f05454ee5ba5f2e5ff0a16c
---
templates/cuda_init.sh.j2 | 9 +++++++++
1 file changed, 9 insertions(+)
create mode 100644 templates/cuda_init.sh.j2
(limited to 'templates/cuda_init.sh.j2')
diff --git a/templates/cuda_init.sh.j2 b/templates/cuda_init.sh.j2
new file mode 100644
index 0000000..3c85ea2
--- /dev/null
+++ b/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
--
cgit v1.2.3