diff --git a/imagesets/generic-worker-win2022/bootstrap.ps1 b/imagesets/generic-worker-win2022/bootstrap.ps1 index a1db8f81..f3ad9c47 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