From 19abf8b92ce332e7566ee537ec3fbdae79c541f5 Mon Sep 17 00:00:00 2001 From: Jillian Rowe Date: Sun, 26 Nov 2023 11:46:46 +0300 Subject: [PATCH] commit --- files/image-builder/ubuntu2004/nvidia-component.yaml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/files/image-builder/ubuntu2004/nvidia-component.yaml b/files/image-builder/ubuntu2004/nvidia-component.yaml index 72770ea..9350d43 100644 --- a/files/image-builder/ubuntu2004/nvidia-component.yaml +++ b/files/image-builder/ubuntu2004/nvidia-component.yaml @@ -66,6 +66,9 @@ phases: sudo apt-get update sudo apt-get -y install cuda nvidia-cuda-dev nvidia-cuda-toolkit + # NVIDIA DRIVER 5.2.5 + sudo apt-get install -y nvidia-driver-525 + # this installs the latest version - 12 # wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb # sudo dpkg -i cuda-keyring_1.0-1_all.deb