From 0daca07a6b6e6005ee75b3bb8539ae3b5abe20b0 Mon Sep 17 00:00:00 2001 From: Ole Markus With Date: Thu, 25 Aug 2022 14:39:55 +0200 Subject: [PATCH] Use specific deb repos based on closest LTS release --- nodeup/pkg/model/nvidia.go | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/nodeup/pkg/model/nvidia.go b/nodeup/pkg/model/nvidia.go index 142e14042f..ddd9b85e3d 100644 --- a/nodeup/pkg/model/nvidia.go +++ b/nodeup/pkg/model/nvidia.go @@ -17,6 +17,8 @@ limitations under the License. package model import ( + "fmt" + "k8s.io/kops/upup/pkg/fi" "k8s.io/kops/upup/pkg/fi/nodeup/nodetasks" ) @@ -31,13 +33,21 @@ var _ fi.ModelBuilder = &NvidiaBuilder{} // Build is responsible for installing packages. func (b *NvidiaBuilder) Build(c *fi.ModelBuilderContext) error { if b.InstallNvidiaRuntime() && b.Distribution.IsUbuntu() { + version := "" + if b.Distribution.Version() >= 22.04 { + version = "22.04" + } else if b.Distribution.Version() >= 20.04 { + version = "20.04" + } else { + version = "18.04" + } c.AddTask(&nodetasks.AptSource{ Name: "nvidia-container-runtime", Keyring: "https://nvidia.github.io/nvidia-container-runtime/gpgkey", Sources: []string{ - "deb https://nvidia.github.io/nvidia-container-runtime/stable/ubuntu18.04/$(ARCH) /", - "deb https://nvidia.github.io/libnvidia-container/stable/ubuntu18.04/$(ARCH) /", - "deb https://nvidia.github.io/nvidia-docker/ubuntu18.04/$(ARCH) /", + fmt.Sprintf("deb https://nvidia.github.io/nvidia-container-runtime/stable/ubuntu%s/$(ARCH) /", version), + fmt.Sprintf("deb https://nvidia.github.io/libnvidia-container/stable/ubuntu%s/$(ARCH) /", version), + fmt.Sprintf("deb https://nvidia.github.io/nvidia-docker/ubuntu%s/$(ARCH) /", version), }, }) c.AddTask(&nodetasks.Package{Name: "nvidia-container-runtime"})