install gpu drivers before extracting hyperkube (#4276)

Azure · Nov 21, 2018 · dab5943 · dab5943
1 parent 53ec5aa
commit dab5943
Showing 1 changed file with 6 additions and 7 deletions.
diff --git 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