From 00e7a074ef77c4b7cad4b7898ffb5254fbbf0090 Mon Sep 17 00:00:00 2001 From: Cameron Meissner Date: Thu, 23 May 2024 10:20:34 -0700 Subject: [PATCH] cherry-pick: Update GRID driver to 535.161.08 for kernel compat issue (#4429) (#4432) Co-authored-by: Ganeshkumar Ashokavardhanan <35557827+ganeshkumarashok@users.noreply.github.com> --- pkg/agent/common/gpu.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/agent/common/gpu.go b/pkg/agent/common/gpu.go index 3a9201f1ba7..e92274e1c71 100644 --- a/pkg/agent/common/gpu.go +++ b/pkg/agent/common/gpu.go @@ -26,13 +26,13 @@ const ( nvidia470CudaDriverVersion = "cuda-470.82.01" nvidia535CudaDriverVersion = "cuda-535.54.03" nvidia550CudaDriverVersion = "cuda-550.54.15" - nvidia535GridDriverVersion = "grid-535.154.05" + nvidia535GridDriverVersion = "grid-535.161.08" ) // These SHAs will change once we update aks-gpu images in aks-gpu repository. We do that fairly rarely at this time. // So for now these will be kept here like this. const ( - aksGPUGridSHA = "sha-13f51b" + aksGPUGridSHA = "sha-d1f0ca" aksGPUCudaSHA = "sha-2d4c96" )