From dab5943279e497d9ccb49938fee5417717bdb5cb Mon Sep 17 00:00:00 2001
From: Jack Francis <jackfrancis@gmail.com>
Date: Wed, 21 Nov 2018 11:47:01 -0800
Subject: [PATCH] install gpu drivers before extracting hyperkube (#4276)

---
 parts/k8s/kubernetescustomscript.sh | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/parts/k8s/kubernetescustomscript.sh b/parts/k8s/kubernetescustomscript.sh
index efa97c100..4ec1cd069 100755
--- a/parts/k8s/kubernetescustomscript.sh
+++ b/parts/k8s/kubernetescustomscript.sh
@@ -73,6 +73,12 @@ fi
 installContainerRuntime
 installNetworkPlugin
 installContainerd
+if [[ "${GPU_NODE}" = true ]]; then
+    if $FULL_INSTALL_REQUIRED; then
+        installGPUDrivers
+    fi
+    ensureGPUDrivers
+fi
 installKubeletAndKubectl
 ensureRPC
 createKubeManifestDir
@@ -122,13 +128,6 @@ if [[ ! -z "${MASTER_NODE}" ]]; then
     ensurePodSecurityPolicy
 fi
 
-if [[ "${GPU_NODE}" = true ]]; then
-    if $FULL_INSTALL_REQUIRED; then
-        installGPUDrivers
-    fi
-    ensureGPUDrivers
-fi
-
 if $FULL_INSTALL_REQUIRED; then
     if [[ $OS == $UBUNTU_OS_NAME ]]; then
         # mitigation for bug https://bugs.launchpad.net/ubuntu/+source/linux/+bug/1676635