chore: Upgrade nvidia drivers to 450.80.02 (#4376)

Azure · Apr 27, 2021 · 86897b5 · 86897b5
1 parent 8d56554
commit 86897b5
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/docs/topics/gpu.md b/docs/topics/gpu.md
@@ -4,7 +4,7 @@ If you created a Kubernetes cluster with one or multiple Ubuntu node pools whose
 
 The NVIDIA drivers are automatically installed on every GPU agent in your cluster, so you don't need to do that manually, unless you require a specific version of the drivers. Currently, the installed driver is version 418.40.04.
 
-Note: You will get version 450.51.06 of the drivers if you are using Ubuntu 18.04-LTS, and version 418.40.04 drivers if you are using 16.04-LTS.
+Note: You will get version 450.80.02 of the drivers if you are using Ubuntu 18.04-LTS, and version 418.40.04 drivers if you are using 16.04-LTS.
 
 To make sure everything is fine, run `kubectl describe node <name-of-a-gpu-node>`. You should see the correct number of GPU reported (in this example shows 2 GPU for a NC12 VM):
 

diff --git a/parts/k8s/cloud-init/artifacts/cse_helpers.sh b/parts/k8s/cloud-init/artifacts/cse_helpers.sh
@@ -12,7 +12,7 @@ if [[ ${OS} == "${UBUNTU_OS_NAME}" ]]; then
 fi
 DOCKER=/usr/bin/docker
 if [[ $UBUNTU_RELEASE == "18.04" ]]; then
-  export GPU_DV=450.51.06
+  export GPU_DV=450.80.02
 else
   export GPU_DV=418.40.04
 fi

diff --git a/pkg/engine/templates_generated.go b/pkg/engine/templates_generated.go
diff --git a/test/e2e/kubernetes/workloads/cuda-vector-add.yaml b/test/e2e/kubernetes/workloads/cuda-vector-add.yaml
@@ -10,7 +10,7 @@ spec:
       restartPolicy: Never
       containers:
       - name: cuda-vector-add
-        image: k8s.gcr.io/cuda-vector-add:v0.1
+        image: nvcr.io/nvidia/k8s/cuda-sample:vectoradd-cuda11.1-ubuntu18.04
       nodeSelector:
         kubernetes.io/os: linux
         accelerator: nvidia