From 63b0d3c18c280e8312eb6c80cf4a021724b3e4d0 Mon Sep 17 00:00:00 2001 From: Mikolaj Umanski Date: Mon, 8 Jan 2024 10:38:23 -0800 Subject: [PATCH] feat: update driver version in dedicated gpu vhd --- vhdbuilder/packer/install-dependencies.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vhdbuilder/packer/install-dependencies.sh b/vhdbuilder/packer/install-dependencies.sh index 8a579379f72..1d4cf6603a3 100644 --- a/vhdbuilder/packer/install-dependencies.sh +++ b/vhdbuilder/packer/install-dependencies.sh @@ -200,8 +200,8 @@ echo " - runc version ${INSTALLED_RUNC_VERSION}" >> ${VHD_LOGS_FILEPATH} if [[ $OS == $UBUNTU_OS_NAME && $(isARM64) != 1 ]]; then # no ARM64 SKU with GPU now gpu_action="copy" - NVIDIA_DRIVER_IMAGE_SHA="sha-16fd35" - export NVIDIA_DRIVER_IMAGE_TAG="cuda-525.85.12-${NVIDIA_DRIVER_IMAGE_SHA}" + NVIDIA_DRIVER_IMAGE_SHA="sha-ff213d" + export NVIDIA_DRIVER_IMAGE_TAG="cuda-535.54.03-${NVIDIA_DRIVER_IMAGE_SHA}" mkdir -p /opt/{actions,gpu} ctr image pull $NVIDIA_DRIVER_IMAGE:$NVIDIA_DRIVER_IMAGE_TAG