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
2018-12-21 11:43:52 -08:00
parent 7e3b54f826
commit 4bf797c8f1
2 changed files with 6 additions and 0 deletions
--- 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
--- a/net/scripts/enable-nvidia-persistence-mode.sh
+++ b/net/scripts/enable-nvidia-persistence-mode.sh
@@ -0,0 +1,2 @@
+#!/usr/bin/env bash
+nvidia-smi -pm ENABLED