Load nvidia drivers on node startup (#2263) (#2264)

* Load nvidia drivers on node startup * added new script to enable nvidia driver persistent mode * remove set -ex
2018-12-21 11:58:06 -08:00
parent 56d5324837
commit 3a039c8007
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