From 4bf797c8f1e43919b6a0d5745d386350991d5866 Mon Sep 17 00:00:00 2001
From: Pankaj Garg <pankaj@solana.com>
Date: Fri, 21 Dec 2018 11:43:52 -0800
Subject: [PATCH] Load nvidia drivers on node startup (#2263)

* Load nvidia drivers on node startup

* added new script to enable nvidia driver persistent mode

* remove set -ex
---
 net/gce.sh                                    | 4 ++++
 net/scripts/enable-nvidia-persistence-mode.sh | 2 ++
 2 files changed, 6 insertions(+)
 create mode 100755 net/scripts/enable-nvidia-persistence-mode.sh

diff --git a/net/gce.sh b/net/gce.sh
index e5b8763293..c29332c9da 100755
--- a/net/gce.sh
+++ b/net/gce.sh
@@ -430,6 +430,10 @@ $(
     network-config.sh \
     remove-docker-interface.sh \
 
+    if "$enableGpu"; then
+      cat enable-nvidia-persistence-mode.sh
+    fi
+
 )
 
 cat > /etc/motd <<EOM
diff --git a/net/scripts/enable-nvidia-persistence-mode.sh b/net/scripts/enable-nvidia-persistence-mode.sh
new file mode 100755
index 0000000000..22bf80032b
--- /dev/null
+++ b/net/scripts/enable-nvidia-persistence-mode.sh
@@ -0,0 +1,2 @@
+#!/usr/bin/env bash
+nvidia-smi -pm ENABLED