From 4e4a93854122b9b6237a0f454cbc5f04fda145e8 Mon Sep 17 00:00:00 2001 From: Pete Moore Date: Wed, 25 Sep 2024 09:52:36 +0200 Subject: [PATCH] Install CUDA on Windows GPU workers --- imagesets/generic-worker-win2022/bootstrap.ps1 | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/imagesets/generic-worker-win2022/bootstrap.ps1 b/imagesets/generic-worker-win2022/bootstrap.ps1 index 4da66d9f..04dfff1d 100644 --- a/imagesets/generic-worker-win2022/bootstrap.ps1 +++ b/imagesets/generic-worker-win2022/bootstrap.ps1 @@ -224,6 +224,10 @@ $hasNvidiaGpu = Get-PnpDevice -PresentOnly | Where-Object { $_.InstanceId -match if ($hasNvidiaGpu) { $client.DownloadFile("https://download.microsoft.com/download/a/3/1/a3186ac9-1f9f-4351-a8e7-b5b34ea4e4ea/538.46_grid_win10_win11_server2019_server2022_dch_64bit_international_azure_swl.exe", "C:\nvidia_driver.exe") Start-Process "C:\nvidia_driver.exe" -ArgumentList "-s", "-noreboot" -Wait -NoNewWindow -RedirectStandardOutput "C:\nvidia-install-stdout.txt" -RedirectStandardError "C:\nvidia-install-stderr.txt" + # install CUDA + # https://github.com/taskcluster/community-tc-config/issues/713 + $client.DownloadFile("https://developer.download.nvidia.com/compute/cuda/12.6.1/local_installers/cuda_12.6.1_560.94_windows.exe", "C:\cuda_installer.exe") + Start-Process "C:\cuda_installer.exe" -ArgumentList "-s", "-noreboot" -Wait -NoNewWindow -RedirectStandardOutput "C:\cuda-install-stdout.txt" -RedirectStandardError "C:\cuda-install-stderr.txt" } # now shutdown, in preparation for creating an image