From 723852a62af669148a6b96ee734ddad05a8e93f0 Mon Sep 17 00:00:00 2001 From: Amit Kumar Date: Thu, 8 Jan 2026 20:51:32 +0000 Subject: [PATCH 1/2] Update CUDA driver --- vm-images/elements/cuda/post-install.d/05-cuda-install | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vm-images/elements/cuda/post-install.d/05-cuda-install b/vm-images/elements/cuda/post-install.d/05-cuda-install index 503c2f6..ab17564 100755 --- a/vm-images/elements/cuda/post-install.d/05-cuda-install +++ b/vm-images/elements/cuda/post-install.d/05-cuda-install @@ -31,7 +31,8 @@ echo 'debconf debconf/frontend select Noninteractive' | sudo debconf-set-selecti DEBIAN_FRONTEND=noninteractive sudo apt-get install -y -q DEBIAN_FRONTEND=noninteractive sudo apt update DEBIAN_FRONTEND=noninteractive sudo apt upgrade -y -DEBIAN_FRONTEND=noninteractive sudo apt install nvidia-driver-525 -y +DEBIAN_FRONTEND=noninteractive sudo apt install -y ubuntu-drivers-common +DEBIAN_FRONTEND=noninteractive sudo ubuntu-drivers install nvidia:580-server echo "Put nvidia drivers on hold: DO NOT UPGRADE basically" dpkg-query -W --showformat='${Package} ${Status}\n' | grep -v deinstall | awk '{ print $1 }' | grep -E 'nvidia.*-[0-9]+$' | xargs -r -L 1 sudo apt-mark hold From dc229a765887fc54ed526a19ae1d7180981617e9 Mon Sep 17 00:00:00 2001 From: Amit Kumar Date: Thu, 15 Jan 2026 08:29:29 +0000 Subject: [PATCH 2/2] Use ubuntu-drivers with --gpgpu flag for server driver installation Update CUDA driver from nvidia-driver-525 to nvidia:580-server using Ubuntu-native packaging. The --gpgpu flag is required for server drivers. Update package hold regex to match new naming pattern (nvidia*580-server, libnvidia*580-server, linux-*-nvidia*580). --- vm-images/elements/cuda/post-install.d/05-cuda-install | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/vm-images/elements/cuda/post-install.d/05-cuda-install b/vm-images/elements/cuda/post-install.d/05-cuda-install index ab17564..0b3af57 100755 --- a/vm-images/elements/cuda/post-install.d/05-cuda-install +++ b/vm-images/elements/cuda/post-install.d/05-cuda-install @@ -32,9 +32,11 @@ DEBIAN_FRONTEND=noninteractive sudo apt-get install -y -q DEBIAN_FRONTEND=noninteractive sudo apt update DEBIAN_FRONTEND=noninteractive sudo apt upgrade -y DEBIAN_FRONTEND=noninteractive sudo apt install -y ubuntu-drivers-common -DEBIAN_FRONTEND=noninteractive sudo ubuntu-drivers install nvidia:580-server +# The --gpgpu flag is required for server drivers (-server suffix) +# See: https://documentation.ubuntu.com/server/how-to/graphics/install-nvidia-drivers/ +DEBIAN_FRONTEND=noninteractive sudo ubuntu-drivers install --gpgpu nvidia:580-server echo "Put nvidia drivers on hold: DO NOT UPGRADE basically" -dpkg-query -W --showformat='${Package} ${Status}\n' | grep -v deinstall | awk '{ print $1 }' | grep -E 'nvidia.*-[0-9]+$' | xargs -r -L 1 sudo apt-mark hold +dpkg-query -W --showformat='${Package} ${Status}\n' | grep -v deinstall | awk '{ print $1 }' | grep -E '(nvidia|libnvidia|linux-.+-nvidia)-.*580' | xargs -r -L 1 sudo apt-mark hold echo "#############################################################" echo "# Install Nvidia Container Toolkit #"