vcpkg/scripts/azure-pipelines/windows/deploy-cudnn.ps1

22 lines
825 B
PowerShell

# Copyright (c) Microsoft Corporation.
# SPDX-License-Identifier: MIT
if (Test-Path -LiteralPath "$PSScriptRoot/utility-prefix.ps1") {
. "$PSScriptRoot/utility-prefix.ps1"
}
$CudnnUrl = Get-AssetUrl `
-InternetUrl 'https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-9.25.0.15_cuda13-archive.zip' `
-BlobAssetName 'cudnn-windows-x86_64-9.25.0.15_cuda13-archive.zip'
[System.IO.DirectoryInfo]$CudnnInstallDir = "$env:ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v13.3"
DownloadAndUnzip -Url $CudnnUrl -Destination $CudnnInstallDir -StripRootDirectory
$CudnnHeaderPath = Join-Path $CudnnInstallDir "include\cudnn.h"
if (Test-Path -LiteralPath $CudnnHeaderPath) {
Write-Host 'cudnn appears correctly installed'
} else {
Write-Error 'cudnn appears broken!'
}