fix: update gpu driver image to support h100s

pkg/agent/baker_test.go

@@ -1734,25 +1734,25 @@ var _ = Describe("Test normalizeResourceGroupNameForLabel", func() {
 
 var _ = Describe("getGPUDriverVersion", func() {
 	It("should use 470 with nc v1", func() {
-		Expect(getGPUDriverVersion("standard_nc6")).To(Equal("cuda-470.82.01"))
+		Expect(getGPUDriverVersion("standard_nc6")).To(Equal(datamodel.Nvidia470CudaDriverVersion))
 	})
 	It("should use 525 cuda with nc v3", func() {
-		Expect(getGPUDriverVersion("standard_nc6_v3")).To(Equal("cuda-525.85.12"))
+		Expect(getGPUDriverVersion("standard_nc6_v3")).To(Equal(datamodel.Nvidia525CudaDriverVersion))
 	})
 	It("should use 535 grid with nv v5", func() {
-		Expect(getGPUDriverVersion("standard_nv6ads_a10_v5")).To(Equal("grid-535.54.03"))
-		Expect(getGPUDriverVersion("Standard_nv36adms_A10_V5")).To(Equal("grid-535.54.03"))
+		Expect(getGPUDriverVersion("standard_nv6ads_a10_v5")).To(Equal(datamodel.Nvidia535GridDriverVersion))
+		Expect(getGPUDriverVersion("Standard_nv36adms_A10_V5")).To(Equal(datamodel.Nvidia535GridDriverVersion))
 	})
 	It("should use 525 cuda with nv v1 (although we don't know if that works)", func() {
-		Expect(getGPUDriverVersion("standard_nv6")).To(Equal("cuda-525.85.12"))
+		Expect(getGPUDriverVersion("standard_nv6")).To(Equal(datamodel.Nvidia525CudaDriverVersion))
 	})
 })
 
 var _ = Describe("getAKSGPUImageSHA", func() {
 	It("should use newest AKSGPUGridSHA with nv v5", func() {
-		Expect(getAKSGPUImageSHA("standard_nv6ads_a10_v5")).To(Equal("sha-20ffa2"))
+		Expect(getAKSGPUImageSHA("standard_nv6ads_a10_v5")).To(Equal(datamodel.AKSGPUGridSHA))
 	})
 	It("should use newest AKSGPUCudaSHA with non grid SKU", func() {
-		Expect(getAKSGPUImageSHA("standard_nc6_v3")).To(Equal("sha-e8873b"))
+		Expect(getAKSGPUImageSHA("standard_nc6_v3")).To(Equal(datamodel.AKSGPUCudaSHA))
 	})
 })

pkg/agent/datamodel/const.go

@@ -129,7 +129,7 @@ const (
 // So for now these will be kept here like this.
 const (
 	AKSGPUGridSHA = "sha-20ffa2"
-	AKSGPUCudaSHA = "sha-e8873b"
+	AKSGPUCudaSHA = "sha-16fd35"
 )
 
 /* ConvergedGPUDriverSizes : these sizes use a "converged" driver to support both cuda/grid workloads.