68 lines
1.6 KiB
PowerShell
68 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)) {
|
|
Write-Host 'Downloading from the Internet'
|
|
$CudaUrl = 'https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_12.8.0_571.96_windows.exe'
|
|
} else {
|
|
Write-Host 'Downloading from vcpkgimageminting using SAS token'
|
|
$SasToken = $SasToken.Replace('"', '')
|
|
$CudaUrl = "https://vcpkgimageminting.blob.core.windows.net/assets/cuda_12.8.0_571.96_windows.exe?$SasToken"
|
|
}
|
|
|
|
# https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
|
|
# Intentionally omitted:
|
|
# demo_suite_12.8
|
|
# documentation_12.8
|
|
# nsight_compute_12.8
|
|
# nsight_systems_12.8
|
|
# nsight_vse_12.8
|
|
# nvvm_samples_12.8
|
|
# visual_studio_integration_12.8
|
|
# visual_profiler_12.8
|
|
# Display.Driver
|
|
DownloadAndInstall -Name 'CUDA' -Url $CudaUrl -Args @(
|
|
'-s',
|
|
'cublas_12.8',
|
|
'cublas_dev_12.8',
|
|
'cuda_profiler_api_12.8',
|
|
'cudart_12.8',
|
|
'cufft_12.8',
|
|
'cufft_dev_12.8',
|
|
'cuobjdump_12.8',
|
|
'cupti_12.8',
|
|
'curand_12.8',
|
|
'curand_dev_12.8',
|
|
'cusolver_12.8',
|
|
'cusolver_dev_12.8',
|
|
'cusparse_12.8',
|
|
'cusparse_dev_12.8',
|
|
'cuxxfilt_12.8',
|
|
'npp_12.8',
|
|
'npp_dev_12.8',
|
|
'nvcc_12.8',
|
|
'nvdisasm_12.8',
|
|
'nvfatbin_12.8',
|
|
'nvjitlink_12.8',
|
|
'nvjpeg_12.8',
|
|
'nvjpeg_dev_12.8',
|
|
'nvml_dev_12.8',
|
|
'nvprof_12.8',
|
|
'nvprune_12.8',
|
|
'nvrtc_12.8',
|
|
'nvrtc_dev_12.8',
|
|
'nvtx_12.8',
|
|
'occupancy_calculator_12.8',
|
|
'opencl_12.8',
|
|
'sanitizer_12.8',
|
|
'thrust_12.8',
|
|
'-n'
|
|
)
|