vcpkg/scripts/azure-pipelines/windows/deploy-cuda.ps1
2024-07-03 20:01:35 -07:00

66 lines
1.6 KiB
PowerShell

# Copyright (c) Microsoft Corporation.
# SPDX-License-Identifier: MIT
param([string]$SasToken)
if (Test-Path "$PSScriptRoot/utility-prefix.ps1") {
. "$PSScriptRoot/utility-prefix.ps1"
}
[string]$CudaUrl
if ([string]::IsNullOrEmpty($SasToken)) {
$CudaUrl = 'https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda_12.5.0_555.85_windows.exe'
} else {
$SasToken = $SasToken.Replace('"', '')
$CudaUrl = "https://vcpkgimageminting.blob.core.windows.net/assets/cuda_12.5.0_555.85_windows.exe?$SasToken"
}
# https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
# Intentionally omitted:
# demo_suite_12.5
# documentation_12.5
# nvvm_samples_12.5
# visual_studio_integration_12.5
# Display.Driver
DownloadAndInstall -Name 'CUDA' -Url $CudaUrl -Args @(
'-s',
'cublas_12.5',
'cublas_dev_12.5',
'cuda_profiler_api_12.5',
'cudart_12.5',
'cufft_12.5',
'cufft_dev_12.5',
'cuobjdump_12.5',
'cupti_12.5',
'curand_12.5',
'curand_dev_12.5',
'cusolver_12.5',
'cusolver_dev_12.5',
'cusparse_12.5',
'cusparse_dev_12.5',
'cuxxfilt_12.5',
'npp_12.5',
'npp_dev_12.5',
'nsight_compute_12.5',
'nsight_systems_12.5',
'nsight_vse_12.5',
'nvcc_12.5',
'nvdisasm_12.5',
'nvfatbin_12.5',
'nvjitlink_12.5',
'nvjpeg_12.5',
'nvjpeg_dev_12.5',
'nvml_dev_12.5',
'nvprof_12.5',
'nvprune_12.5',
'nvrtc_12.5',
'nvrtc_dev_12.5',
'nvtx_12.5',
'occupancy_calculator_12.5',
'opencl_12.5',
'sanitizer_12.5',
'thrust_12.5',
'visual_profiler_12.5',
'-n'
)